How to use dxvyaaa/yolo_helmet with ultralytics:
from ultralytics import YOLOvv8 model = YOLOvv8.from_pretrained("dxvyaaa/yolo_helmet") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)
How to use dxvyaaa/yolo_helmet with YOLOv10:
This is a YOLOv8 model trained for helmet detection.
from ultralytics import YOLOvv8 model = YOLOvv8.from_pretrained("dxvyaaa/yolo_helmet") source = 'http://images.cocodataset.org/val2017/000000039769.jpg' model.predict(source=source, save=True)