Helmet Detection_YOLOv8 Computer Vision Project

Learning Evidence

Updated a year ago

903

views

65

downloads
Classes (2)
With Helmet
Without Helmet

Metrics

Try This Model
Drop an image or
Description

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

Use This Trained Model

Try it in your browser, or deploy via our Hosted Inference API and other deployment methods.

Supervision

Build Computer Vision Applications Faster with Supervision

Visualize and process your model results with our reusable computer vision tools.

Cite This Project

LICENSE
CC BY 4.0

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-12-25 },
                            }
                        
                    

Similar Projects

See More