--- license: agpl-3.0 language: - en metrics: - precision - accuracy - f1 - recall tags: - object-detection - computer-vision - yolo - yolov11 - helmet-detection - safety - traffic-monitoring - cctv pipeline_tag: object-detection library_name: ultralytics base_model: - Ultralytics/YOLO11 --- # Helmet and No-Helmet Detection — YOLO11s A YOLO11s object detection model trained for detecting **helmets** and **no-helmet instances** in images. The training dataset was obtained from Roboflow and was originally created by another Roboflow user. The dataset was forked and used for training this model. Dataset attribution and licensing information are provided below. ## Model Description This model is a custom-trained **Ultralytics YOLO11s** object detection model. The model predicts two classes: | Class ID | Class | |----------|-------| | 0 | helmet | | 1 | no-helmet | The model accepts images at a nominal resolution of **640 × 640 pixels** and produces bounding-box detections for the two classes. ### Model Architecture - Architecture: YOLO11s - Task: Object Detection - Parameters: 9,428,566 - Layers: 181 - GFLOPs: 21.6 - Input image size: 640 × 640 - Number of classes: 2 The checkpoint identifies the architecture as YOLO11s and reports 181 layers, 9,428,566 parameters, and 21.6 GFLOPs. ## Intended Use This model is intended for research, experimentation, and computer vision applications involving helmet compliance detection. Potential applications include: - Helmet detection in CCTV footage - Road safety monitoring - Traffic violation detection - Industrial safety monitoring - Motorcycle helmet compliance analysis - Computer vision research The model is particularly intended as a component of a larger computer vision pipeline rather than as a complete traffic-violation system. For example: ```text CCTV Image | v Helmet Detector | +---- helmet | +---- no-helmet | v Person / Vehicle Association | v Number Plate Detection | v Violation Processing ``` This model itself only performs helmet/no-helmet object detection. ## Classes The class mapping stored in the trained checkpoint is: 0: helmet 1: no-helmet The model should therefore be interpreted using this class mapping when processing its predictions. ## Training The model was trained using a dataset exported from Roboflow. The original dataset was not created by the author of this model. Instead, the dataset was forked from an existing Roboflow dataset and subsequently used for training. The checkpoint stores the following training configuration. ### Training Configuration | Parameter | Value | |-----------|-------| | Task | Detection | | Image size | 640 | | Batch size | 16 | | Epochs configured | 100 | | Pretrained | True | | Optimizer | Auto | | Workers | 8 | | AMP | True | | Seed | 0 | | Deterministic | True | | Patience | 10 | | Validation | True | | Validation split | val | The checkpoint was produced using Ultralytics version 8.3.233. The stored checkpoint metadata identifies the model as an `ultralytics.nn.tasks.DetectionModel`. ## Data Augmentation The stored training configuration includes the following augmentation settings: | Augmentation | Value | |--------------|-------| | Mosaic | 1.0 | | MixUp | 0.12 | | Copy-Paste | 0.05 | | Horizontal Flip | 0.5 | | Scale | 0.6 | | Rotation | 4.0 | | Translation | 0.1 | | Shear | 1.0 | | Perspective | 0.0004 | These values are reported from the training configuration stored inside the model checkpoint. ## Dataset ### Original Dataset The training dataset was obtained from Roboflow. **Original dataset:** [![Helmet-detection_yolov8-8jenr](https://img.shields.io/badge/helmet%20detection-yolov8%208jenr(Dataset)-00000)](https://app.roboflow.com/smolry/helmet-detection_yolov8-8jenr/1/images) **Original dataset author:** [![Smolry](https://img.shields.io/badge/Smolry-00000)](https://app.roboflow.com/smolry) **Roboflow project:** [![helmet-detection_yolov8-8jenr](https://img.shields.io/badge/helmet%20detection-yolov8%208jenr-00000)](https://app.roboflow.com/smolry/helmet-detection_yolov8-8jenr) **Dataset version:** [v1 2026-01-25 2:39am] The dataset was forked from the original Roboflow project and used as the basis for training this model. ### Dataset Attribution This model does not claim ownership of the original dataset. The dataset and its annotations remain subject to the original dataset's license and attribution requirements. Users of this model should consult the original dataset page and license before redistributing the dataset, annotations, or derived datasets. ## Data Preprocessing The model was trained using the YOLO-compatible dataset configuration exported from Roboflow. The checkpoint references the following dataset configuration: /content/Helmet-and-Non-Helmet-Detection--2/data.yaml The original training environment was hosted in Google Colab / Google Drive according to paths recorded in the checkpoint. ## Model Input The model expects an image input and was trained using: 640 × 640 Ultralytics handles the necessary image preprocessing during normal inference. ## Inference Install Ultralytics: pip install ultralytics Load the model: ```python from ultralytics import YOLO model = YOLO("best.pt") results = model("image.jpg", imgsz=640) for result in results: result.show()