predict 5 classes
- "names": ['hat','nohat','novest','person','vest']
Performance
- yolo11n series : mAP50 0.708
- yolo11m series : mAP50 0.809
Showcase
showcase
Usage
from ultralytics import YOLO
import cv2
model = YOLO("yolo11m_safety.pt")
img= r"test.jpeg"
results = model(img)
predicted = results[0].plot()
cv2.imwrite('predicted.jpg', predicted)
