Detect The Traffic Sign Using YoloV5
Data Collected from Roboflow website.
- Total 3708 traffic sign images are present in 43 classes .
- Total 2538 images for training and 701 images for validation present in 2 classes.
- Create bounding boxes with the help of makesense.ai website.
-
Prepare folder structure that can be accept by YoloV5.
-
Total 3708 images for training and 352 images for validation present in 43 classes.
-
Create a bounding boxes with the help of label-img And makesense.ai website according to YoloV5.
-
Cloning the YoloV5 file from official repository.
-
Changing the directory of yolov5
-
Installing the dependencies
-
Download all versions pre-trained weights
-
Go to yolov5/data/
-
Open coco128.yaml
-
Edit the following inside it:
A. Training and Validation file path
B. Number of classes and Class names.
- Set images size 408 with batch of 8
- Train model on 1000 epochs
- Gives the data file path as well as give pre-trained weights path.
VISUALISE THE TRAINING METRICS WITH THE HELP OF TENSORBOARD
