YOLOv10

Instalación

pip install git+https://github.com/THU-MIG/yolov10.git

Entrenamiento y Validación

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained('jameslahm/yolov10l')
# Entrenamiento
model.train(...)
# después del entrenamiento, se puede subir al hub
model.push_to_hub("your-hf-username/yolov10-finetuned")

# Validación
model.val(...)

Inferencia

Aquí hay un ejemplo de extremo a extremo que muestra la inferencia en una imagen de gatos:

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained('jameslahm/yolov10l')
source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
model.predict(source=source, save=True)