ppe-detection-yolov8m

PPE Detection — YOLOv8m

Fine-tuned YOLOv8m for construction-site personal protective equipment (PPE) compliance. Detects PPE worn and, critically, PPE violations (missing hardhat, mask, safety vest, gloves, goggles).

Used as a built-in pattern-alert detector in the visual-ai platform; downloaded automatically at runtime via huggingface_hub.hf_hub_download.

Classes (14)

0  Fall-Detected     5  Mask          10  NO-Safety Vest
1  Gloves            6  NO-Gloves     11  Person
2  Goggles           7  NO-Goggles    12  Safety Cone
3  Hardhat           8  NO-Hardhat    13  Safety Vest
4  Ladder            9  NO-Mask

Violation (alert) classes: NO-Hardhat, NO-Mask, NO-Safety Vest, NO-Gloves, NO-Goggles.

Metrics (validation, best checkpoint, 22,077 instances)

Per-class recall / mAP@50 for the violation classes:

Note: NO-Safety Vest is the rarest class and has weak recall — treat its alerts with lower confidence and consider oversampling/more data for a future revision.

Training

• Base: yolov8m.pt, 640px, batch 16, ~92/100 epochs (early stop, patience 25), seed 0
• GPU: RTX 4070 (12 GB), CUDA 12.1

Usage

from huggingface_hub import hf_hub_download
from ultralytics import YOLO

path = hf_hub_download(repo_id="jashwanthpeddisetty0712/ppe-detection-yolov8m", filename="best.pt")
model = YOLO(path)
results = model.predict("image.jpg", conf=0.25)