# traffic_sign_classification

**Repository Path**: Raiden_cn/traffic_sign_classification

## Basic Information

- **Project Name**: traffic_sign_classification
- **Description**: A baseline for traffic sign classification
交通标识识别baseline
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2022-09-04
- **Last Updated**: 2022-09-06

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 交通标志识别大作业

# Environment

pytorch >= 1.8
timm
tensorboardX
sklearn
albumentations

only test in pytorch=1.8

# Directory Structure

├── ckpt  
│   └── swin_model_100_0.977.pth  
├── data  
│   ├── GuideSign  
│   ├── M1  
│   ├── M4  
│   ├── M5  
│   ├── M6  
│   ├── M7  
│   ├── P1  
│   ├── P10_50  
│   ├── P12  
│   └── W1  
├── data4test (only jpg files)  
├── data4train  
│   ├── train  
│   │   ├── GuideSign  
│   │   ├── M1 
│   │   ├── M4  
│   │   ├── M5  
│   │   ├── M6  
│   │   ├── M7  
│   │   ├── P1  
│   │   ├── P10_50  
│   │   ├── P12    
│   │   └── W1  
│   └── val  
│       ├── GuideSign  
│       ├── M1  
│       ├── M4  
│       ├── M5  
│       ├── M6  
│       ├── M7  
│       ├── P1  
│       ├── P10_50  
│       ├── P12  
│       └── W1  
├── makedata.py  
├── README.md  
├── submit_json  
│   └── swin_transformer.json  
├── test.py  
├── train.py  
└── val_one_ckpt.py  



## makedata.py

generating trianval split

1. set original path and new path in line8 and line10

2. run 

   ```python
   python makedata.py
   ```

   before running

   > ├── data  
   > │   ├── GuideSign  
   > │   ├── M1  
   > │   ├── M4  
   > │   ├── M5  
   > │   ├── M6  
   > │   ├── M7  
   > │   ├── P1  
   > │   ├── P10_50  
   > │   ├── P12  
   > │   └── W1  
   > ├── data4test  
   > ├── data4train  

   after running  

   > ├── data  
   > │   ├── GuideSign  
   > │   ├── M1  
   > │   ├── M4  
   > │   ├── M5  
   > │   ├── M6  
   > │   ├── M7  
   > │   ├── P1  
   > │   ├── P10_50  
   > │   ├── P12  
   > │   └── W1  
   > ├── data4test  
   > ├── data4train  
   > │   ├── train  
   > │   │   ├── GuideSign  
   > │   │   ├── M1  
   > │   │   ├── M4  
   > │   │   ├── M5  
   > │   │   ├── M6  
   > │   │   ├── M7  
   > │   │   ├── P1  
   > │   │   ├── P10_50  
   > │   │   ├── P12  
   > │   │   └── W1  
   > │   └── val  
   > │       ├── GuideSign  
   > │       ├── M1  
   > │       ├── M4  
   > │       ├── M5  
   > │       ├── M6  
   > │       ├── M7  
   > │       ├── P1  
   > │       ├── P10_50  
   > │       ├── P12  
   > │       └── W1  

## train.py

1. Setting parameters for train

2. run 

   ```python
   python train.py
   ```

## test.py

1. Setting parameters for test

2. run

   ```python
   python test.py
   ```

## val_one_ckpt.py

validate specified ckpt.