baskarmother
/

yolov8s-ppe-construction-v2

Object Detection

construction-safety

Model card Files Files and versions

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

yolov8s-ppe-construction-v2

Improved PPE Compliance Detection Model for Construction Sites (v2)

Classes (18)

0: person
1: helmet
2: vest
3: mask
4: gloves
5: safety_shoe
6: goggles
7: no_helmet
8: no_mask
9: no_vest
10: head
11: barricade
12: dumpster
13: excavators
14: safety_net
15: dump_truck
16: truck
17: wheel_loader

Usage

from ultralytics import YOLO
model = YOLO("hf://baskarmother/yolov8s-ppe-construction-v2/best.pt")
results = model.predict("image.jpg")

Training Details

Base Model: YOLOv8s
Epochs: 150
Image Size: 640x640
Batch Size: 16

Generated by ML Intern

This model repository was generated by ML Intern, an agent for machine learning research and development on the Hugging Face Hub.

Try ML Intern: https://smolagents-ml-intern.hf.space
Source code: https://github.com/huggingface/ml-intern

Downloads last month: 5