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:
$ pip install -r requirements.txt
Tutorials
- Trainning by google colab 🚀 RECOMMENDED
Data preparation
$ wget https://github.com/makerhanoi/via-datasets/releases/download/v1.0/via-trafficsign-20210321.zip
$ unzip via-trafficsign-20210321.zip
Training
$ 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.