Instructions to use dxvyaaa/yolo_helmet with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- ultralytics
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) - YOLOv10
How to use dxvyaaa/yolo_helmet with YOLOv10:
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) - Notebooks
- Google Colab
- Kaggle
| # data.yaml | |
| path: . | |
| train: images | |
| val: images | |
| nc: 2 | |
| names: | |
| 0: helmet | |
| 1: head | |