Traffic Sign Detection Using YoloV5

Alt text

Instalation

clone the repository:

$ git clone https://github.com/maheravi/YoloV5-TrafficSign.git

%cd YoloV5-TrafficSign

!pip install -U -r requirements.txt

Dataset

The dataset provided by Aryan Garg.

You can download it form the link below or downloading usinge the code in YoloV5_Traffic_Sign.ipynb

Dataset link

Train

!python train.py --img 640 --batch 16 --epochs 50 --data dataset.yaml --weights yolov5s.pt

Test

!python val.py --data dataset.yaml --weights /content/yolov5/runs/train/exp/best.pt --img 640

Inference

!python detect.py --source /content/yolov5/Sign.jpg --weights /content/yolov5/runs/train/exp/best.pt

GitHub

View Github