from ultralytics import YOLO

# Load from Hugging Face
model = YOLO("hf://Hexmon/vyra-yolo-ppe-detection/best.pt")

# Run inference
results = model.predict("image.jpg", conf=0.5)
results[0].show()

Training Details

Base model: YOLOv8m (pretrained on COCO)
Image size: 640
Epochs: 100
Batch size: 32
Dataset: PPE Combined Model v4 (CC BY 4.0)

Usage Examples

Webcam:

model.predict(source=0, show=True, conf=0.5)

Video:

model.predict("video.mp4", conf=0.5, save=True)

Access detections:

python

results = model.predict("image.jpg", conf=0.5)
for box in results[0].boxes:
    print(model.names[int(box.cls)], float(box.conf))