This project uses object detection models: Yolov5, Yolov7, and DeepSORT. It classifies motorcycle riders and helmet types from video input.
In the first .ipynb file (HelmetDetection_YOLOv5.ipynb) It used YOLOv5 and the model performance analysis can be seen in here
Image from the video output
(Video Link Here)