wesjos/Yolo-hard-hat-safety-vest

predict 5 classes

• "names": ['hat','nohat','novest','person','vest']

Performance

• yolo11n series : mAP50 0.708
• yolo11m series : mAP50 0.809

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)