Update README.md
Browse files
README.md
CHANGED
@@ -1,9 +1,47 @@
|
|
1 |
---
|
|
|
2 |
tags:
|
3 |
-
-
|
4 |
-
-
|
|
|
|
|
|
|
5 |
---
|
6 |
|
7 |
-
|
8 |
-
-
|
9 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
license: agpl-3.0
|
3 |
tags:
|
4 |
+
- object-detection
|
5 |
+
- computer-vision
|
6 |
+
- yolov10
|
7 |
+
datasets:
|
8 |
+
- detection-datasets/coco
|
9 |
---
|
10 |
|
11 |
+
### Model Description
|
12 |
+
[YOLOv10: Real-Time End-to-End Object Detection](https://arxiv.org/abs/2405.14458v1)
|
13 |
+
|
14 |
+
- arXiv: https://arxiv.org/abs/2405.14458v1
|
15 |
+
- github: https://github.com/THU-MIG/yolov10
|
16 |
+
|
17 |
+
### Installation
|
18 |
+
```
|
19 |
+
pip install supervision git+https://github.com/THU-MIG/yolov10.git
|
20 |
+
```
|
21 |
+
|
22 |
+
### Yolov10 Training/Validation/Prediction
|
23 |
+
```python
|
24 |
+
from ultralytics import YOLOv10
|
25 |
+
|
26 |
+
model = YOLOv10.from_pretrained('jameslahm/yolov10b')
|
27 |
+
# Training
|
28 |
+
model.train(...)
|
29 |
+
# after training, one can push to the hub
|
30 |
+
model.push_to_hub("your-hf-username/yolov10-finetuned")
|
31 |
+
|
32 |
+
# Validation
|
33 |
+
model.val(...)
|
34 |
+
|
35 |
+
# Prediction
|
36 |
+
model.predict(...)
|
37 |
+
```
|
38 |
+
|
39 |
+
### BibTeX Entry and Citation Info
|
40 |
+
```
|
41 |
+
@article{wang2024yolov10,
|
42 |
+
title={YOLOv10: Real-Time End-to-End Object Detection},
|
43 |
+
author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
|
44 |
+
journal={arXiv preprint arXiv:2405.14458},
|
45 |
+
year={2024}
|
46 |
+
}
|
47 |
+
```
|