An AI based helmet detection , which can detect the helmets from the images as well as videos It is implemented using Opencv, Python, YOLOV5
- Took dataset from kaggle : https://www.kaggle.com/andrewmvd/helmet-detection
- As yolov5 requires annotations in txt format, converted xml to txt
- Applied data agumentations using albumentations
- Trained for 30 epochs