Downloads · 30 days
959
2% of all-time downloads
Hexmon/vyra-yolo-ppe-detection
vyra-yolo-ppe-detection is a object detection model from Hexmon. Use it when you need objects located in an image. It is set up for ultralytics. The card lists the license as cc-by-4.0.
A YOLOv8-medium model fine-tuned to detect Personal Protective Equipment (PPE) in images and video.
Downloads · 30 days
959
2% of all-time downloads
All-time downloads
49.6K
Public
Repo size
156 MB
Likes
9
Public
Click a slice to open those files.
.onnx104 MB · 66%
From the Hugging Face model README
A YOLOv8-medium model fine-tuned to detect Personal Protective Equipment (PPE) in images and video.
| ID | Class | ID | Class |
|---|---|---|---|
| 0 | Fall-Detected | 7 | NO-Goggles |
| 1 | Gloves | 8 | NO-Hardhat |
| 2 | Goggles | 9 | NO-Mask |
| 3 | Hardhat | 10 | NO-Safety Vest |
| 4 | Ladder | 11 | Person |
| 5 | Mask | 12 | Safety Cone |
| 6 | NO-Gloves | 13 | Safety Vest |
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()
Webcam:
model.predict(source=0, show=True, conf=0.5)
Video:
model.predict("video.mp4", conf=0.5, save=True)
Access detections:
results = model.predict("image.jpg", conf=0.5)
for box in results[0].boxes:
print(model.names[int(box.cls)], float(box.conf))