/Helmet-Detection

Helmet Detection using YOLOv5s

Primary LanguageJupyter Notebook

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)

Alt Text Alt Text