|
--- |
|
license: agpl-3.0 |
|
tags: |
|
- object-detection |
|
- computer-vision |
|
- yolov10 |
|
datasets: |
|
- detection-datasets/coco |
|
--- |
|
|
|
### Model Description |
|
[YOLOv10: Real-Time End-to-End Object Detection](https://arxiv.org/abs/2405.14458v1) |
|
|
|
- arXiv: https://arxiv.org/abs/2405.14458v1 |
|
- github: https://github.com/THU-MIG/yolov10 |
|
|
|
### Installation |
|
``` |
|
pip install supervision git+https://github.com/THU-MIG/yolov10.git |
|
``` |
|
|
|
### Yolov10 Training/Validation/Prediction |
|
```python |
|
from ultralytics import YOLOv10 |
|
|
|
model = YOLOv10.from_pretrained('jameslahm/yolov10s') |
|
# Training |
|
model.train(...) |
|
# after training, one can push to the hub |
|
model.push_to_hub("your-hf-username/yolov10-finetuned") |
|
|
|
# Validation |
|
model.val(...) |
|
|
|
# Prediction |
|
model.predict(...) |
|
``` |
|
|
|
### 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} |
|
} |
|
``` |
