This repo train YOLOv5 on the dataset of trafficsigns, including 11095 images, 6 classes from Goodgame and ICT teams - FPT Digital Race.

Requirements

Python 3.8 or later with all requirements.txt dependencies installed, including torch>=1.7. To install run:

1

$ pip install -r requirements.txt

Tutorials

• Trainning by google colab  🚀 RECOMMENDED

Data preparation

1
2

$ wget https://github.com/makerhanoi/via-datasets/releases/download/v1.0/via-trafficsign-20210321.zip
$ unzip via-trafficsign-20210321.zip

Training

1
2

$ python train.py --img 224 --batch 16 --epochs 30 --data via-trafficsign.yaml\
 --weights yolov5s.pt --hyp hyp.yaml

Pretrained model

• The pretrained YOLOv5s on drive.