1
0
Fork 0
Go to file
ailab 093a6e0361 first commit 2024-07-01 11:23:22 +08:00
.gitattributes first commit 2024-07-01 11:23:22 +08:00
README.md first commit 2024-07-01 11:23:22 +08:00
config.json first commit 2024-07-01 11:23:22 +08:00
model.safetensors first commit 2024-07-01 11:23:22 +08:00

README.md

license tags datasets inference
agpl-3.0
object-detection
computer-vision
yolov10
detection-datasets/coco
false

Model Description

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

Installation

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

Training and validation

from ultralytics import YOLOv10

model = YOLOv10.from_pretrained('jameslahm/yolov10l')
# Training
model.train(...)
# after training, one can push to the hub
model.push_to_hub("your-hf-username/yolov10-finetuned")

# Validation
model.val(...)

Inference

Here's an end-to-end example showcasing inference on a cats image:

from ultralytics import YOLOv10

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

which shows:

image/png

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}
}