yolov10s / README.md
jameslahm's picture
Add code snippet regarding push to hub (#1)
5ef34f6 verified
|
raw
history blame
1.03 kB
metadata
license: agpl-3.0
tags:
  - object-detection
  - computer-vision
  - yolov10
datasets:
  - detection-datasets/coco

Model Description

YOLOv10: Real-Time End-to-End Object Detection

Installation

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

Yolov10 Inference

from ultralytics import YOLOv10
import supervision as sv
import cv2

IMAGE_PATH = 'dog.jpeg'

model = YOLOv10.from_pretrained('jameslahm/yolov10s')
model.predict(IMAGE_PATH, show=True)

# after training, one can push to the hub
model.push_to_hub("your-hf-username/yolov10-finetuned")

BibTeX Entry and Citation Info

@article{wang2024yolov10,
 title={YOLOv10: Real-Time End-to-End Object Detection},
 author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
 journal={arXiv preprint arXiv:2405.14458},
 year={2024}
}