Bike Helmet Detection Computer Vision Project

Bike Helmets

Updated 3 years ago

42k

views

2.2k

downloads
Classes (2)
With Helmet
Without Helmet

Metrics

Try This Model
Drop an image or
Description

Background Information

This dataset was curated and annotated by Syed Salman Reza. A custom dataset composed of two classes (With Helmet, Without Helmet). Main objetive is to identify if a Biker wearing Helmet or not.

The original custom dataset (v1) is composed of 1,371 images of people with and without bike helmets.

The dataset is available under the Public License.

Example of an Annotated Image from the Dataset

Getting Started

You can download this dataset for use within your own projects, or fork it into a workspace on Roboflow to create your own model.

Dataset Versions

Version 1 (v1) - 1,371 images

  • Preprocessing: Auto-Orient and Resize (Stretch to 416x416)
  • Augmentations: Augmentations applied prior to import - Bounding Box Blur (up to 10px)
  • Training Metrics: Trained from the COCO Checkpoint in Public Models ("transfer learning") on Roboflow
    • mAP = 74.4%, precision = 54.0%, recall = 77.0%

Version 2 (v2) - 3,735 images

  • Preprocessing: Auto-Orient and Resize (Stretch to 416x416)
  • Augmentations: Augmentations applied prior to import - Bounding Box Blur.
    • New augmentations: Outputs per training example: 3 Rotation: Between -30° and +30° Shear: ±15° Horizontal, ±15° Vertical Blur: Up to 1.5px Mosaic: Applied
  • Training Metrics: Trained from the COCO Checkpoint in Public Models ("transfer learning") on Roboflow
    • mAP = 91.5%, precision = 65.1%, recall = 92.8%

Syed Salman Reza - Github

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
Public Domain

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

                        @misc{
                            bike-helmet-detection-2vdjo_dataset,
                            title = { Bike Helmet Detection Dataset },
                            type = { Open Source Dataset },
                            author = { Bike Helmets },
                            howpublished = { \url{ https://universe.roboflow.com/bike-helmets/bike-helmet-detection-2vdjo } },
                            url = { https://universe.roboflow.com/bike-helmets/bike-helmet-detection-2vdjo },
                            journal = { Roboflow Universe },
                            publisher = { Roboflow },
                            year = { 2021 },
                            month = { sep },
                            note = { visited on 2024-12-18 },
                            }