YOLOv8 PPE Compliance Detection for Construction Sites
This model detects Personal Protective Equipment (PPE) compliance on construction sites.
Classes (17 categories)
- 0: barricade
- 1: dumpster
- 2: excavators
- 3: gloves
- 4: hardhat
- 5: mask
- 6: no-hardhat
- 7: no-mask
- 8: no-safety vest
- 9: person
- 10: safety net
- 11: safety shoes
- 12: safety vest
- 13: dump truck
- 14: mini-van
- 15: truck
- 16: wheel loader
Training Details
- Base Model: yolov8n
- Dataset: keremberke/construction-safety-object-detection
- Image Size: 640x640
- Epochs: 100
- Optimizer: SGD (lr=0.01, momentum=0.9)
Usage
from ultralytics import YOLO
from huggingface_hub import hf_hub_download
model = YOLO(hf_hub_download("baskarmother/yolov8-ppe-construction", "best.pt"))
results = model("your_image.jpg")
results[0].plot()
- Downloads last month
- 40