Images

1737 images

Explore Dataset

This project focuses on implementing a real-time helmet detection system using the YOLO v8 model. The researchers utilized two image datasets sourced from Kaggle, both containing annotated images specifically for helmet detection. These datasets primarily facilitate the binary classification of helmet presence, categorizing images into "With helmet" and "Without helmet" classes.

Dataset Insights

During the exploration phase, it was observed that the datasets exhibit data imbalance, showcasing varying counts between images depicting helmets and those without helmets. Recognizing this imbalance is crucial as it may impact the YOLO v8 model's performance during training. Understanding these key characteristics of the datasets forms the cornerstone of our approach in subsequent phases. This understanding enables us to address challenges related to class imbalance with the use of data augmentation techniques and ensures the robust training of the model for effective real-time helmet detection within the domain of traffic monitoring and surveillance.

Contributors

Ken C. Aquitan

Christian A. Muaña

Dawn Angela R. Velasquez

Trained Model API

This project has a trained model available that you can try in your browser and use to get predictions via our Hosted Inference API and other deployment methods.

YOLOv8

This project has a YOLOv8 model checkpoint available for inference with Roboflow Deploy. YOLOv8 is a new state-of-the-art real-time object detection model.

Cite This Project

If you use this dataset in a research paper, please cite it using the following BibTeX:

@misc{
                            helmet-detection_yolov8_dataset,
                            title = { Helmet Detection_YOLOv8 Dataset },
                            type = { Open Source Dataset },
                            author = { Learning Evidence },
                            howpublished = { \url{ https://universe.roboflow.com/learning-evidence/helmet-detection_yolov8 } },
                            url = { https://universe.roboflow.com/learning-evidence/helmet-detection_yolov8 },
                            journal = { Roboflow Universe },
                            publisher = { Roboflow },
                            year = { 2024 },
                            month = { jan },
                            note = { visited on 2024-04-28 },
                            }