alexwww94/yolov8

This is a Hugging Face Hub model repository created for ultraytics; you still need to install ultraytics(https://github.com/ultralytics/ultralytics)

pip install ultraytics

Usage:

from transformers import AutoModelForObjectDetection
model = AutoModelForObjectDetection.from_pretrained('alexwww94/yolov8', trust_remote_code=True, yolo_model_config={"model": 'yolov8n'})
model('example.jpg')

Parameters for creating the yolov8 model are defined in yolo_model_config. Inference parameters for yolov8 can also be defined using args and kwargs:

model(*args, **kwargs)