Traffic Sign Detection Using YoloV5

Instalation

clone the repository:

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

%cd YoloV5-TrafficSign

!pip install -U -r requirements.txt

The dataset provided by Aryan Garg.

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

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

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

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

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
data		data
models		models
utils		utils
.dockerignore		.dockerignore
.gitattributes		.gitattributes
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
CONTRIBUTING.md		CONTRIBUTING.md
Dockerfile		Dockerfile
LICENSE		LICENSE
README.md		README.md
Sign.jpg		Sign.jpg
YoloV5_Traffic_Sign.ipynb		YoloV5_Traffic_Sign.ipynb
dataset.yaml		dataset.yaml
detect.py		detect.py
export.py		export.py
hubconf.py		hubconf.py
requirements.txt		requirements.txt
setup.cfg		setup.cfg
train.py		train.py
tutorial.ipynb		tutorial.ipynb
val.py		val.py