Links
Abstract
Coded in python using object detection models: Yolov5, Yolov7, and DeepSORT. Classified motorcycle riders and helmet types from video input. The purpose of the project is to automatically detect if motorcycle riders are using the correct helmet types from CCTV footage.
Sample Video Output
Related material
- YOLOv5 – YOLOv5 repository
- YOLOv5 Documentation - Documentation of YOLOv5 model