Push model using huggingface_hub.
Browse files- README.md +5 -38
- config.json +5 -1
README.md
CHANGED
@@ -1,42 +1,9 @@
|
|
1 |
---
|
2 |
-
license: agpl-3.0
|
3 |
tags:
|
4 |
-
-
|
5 |
-
-
|
6 |
-
- yolov10
|
7 |
-
datasets:
|
8 |
-
- detection-datasets/coco
|
9 |
---
|
10 |
|
11 |
-
|
12 |
-
|
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 Inference
|
23 |
-
```python
|
24 |
-
from ultralytics import YOLOv10
|
25 |
-
import supervision as sv
|
26 |
-
import cv2
|
27 |
-
|
28 |
-
IMAGE_PATH = 'dog.jpeg'
|
29 |
-
|
30 |
-
model = YOLOv10.from_pretrained('jameslahm/yolov10m')
|
31 |
-
model.predict(IMAGE_PATH, show=True)
|
32 |
-
```
|
33 |
-
|
34 |
-
### BibTeX Entry and Citation Info
|
35 |
-
```
|
36 |
-
@article{wang2024yolov10,
|
37 |
-
title={YOLOv10: Real-Time End-to-End Object Detection},
|
38 |
-
author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
|
39 |
-
journal={arXiv preprint arXiv:2405.14458},
|
40 |
-
year={2024}
|
41 |
-
}
|
42 |
-
```
|
|
|
1 |
---
|
|
|
2 |
tags:
|
3 |
+
- pytorch_model_hub_mixin
|
4 |
+
- model_hub_mixin
|
|
|
|
|
|
|
5 |
---
|
6 |
|
7 |
+
This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
|
8 |
+
- Library: [More Information Needed]
|
9 |
+
- Docs: [More Information Needed]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
config.json
CHANGED
@@ -1,3 +1,7 @@
|
|
1 |
{
|
2 |
-
"
|
|
|
|
|
|
|
|
|
3 |
}
|
|
|
1 |
{
|
2 |
+
"args": null,
|
3 |
+
"model": "yolov10m.pt",
|
4 |
+
"names": null,
|
5 |
+
"task": null,
|
6 |
+
"verbose": false
|
7 |
}
|