├── README.md
├── Trafficsigns_yolov5.ipynb
└── dataset.yaml


/README.md:
--------------------------------------------------------------------------------
 1 | # Traffic-Sign-Detection-Using-Yolov5
 2 |  Yolov5 to train our custom object detection model.
 3 |  
 4 |  official YOLOv5 notebook by Ultralytic :https://www.kaggle.com/ultralytics/yolov5/notebook
 5 |  
 6 |  Github: https://github.com/ultralytics/yolov5
 7 |  
 8 |  DataSet: https://www.kaggle.com/valentynsichkar/traffic-signs-dataset-in-yolo-format
 9 |  
10 |  HOW TO: Install PyTorch (with GPU) in Windows 10 : https://www.youtube.com/watch?v=eodnCUzSeTk
11 | 


--------------------------------------------------------------------------------
/dataset.yaml:
--------------------------------------------------------------------------------
1 | # train and val data
2 | train: ../images/train/
3 | val: ../images/validation/
4 | 
5 | # number of classes
6 | nc: 4
7 | 
8 | # class names
9 | names: ['speed limit', 'yield', 'mandatory', 'other']


--------------------------------------------------------------------------------