IMvision12 / Helmet-Detection

Helmet Detection using YOLOv5s

Helmet-Detection using YOLOv5

An AI based helmet detection , which can detect the helmets from the images as well as videos It is implemented using Opencv, Python, YOLOV5

Dataset

1. Took dataset from kaggle : https://www.kaggle.com/andrewmvd/helmet-detection
2. As yolov5 requires annotations in txt format, converted xml to txt
3. Applied data agumentations using albumentations
4. Trained for 30 epochs

mAP score : 0.76

Results : (Full video in output folder)