jameslahm commited on
Commit
2c0403e
1 Parent(s): 1a63db7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -5
README.md CHANGED
@@ -1,9 +1,47 @@
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]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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/yolov10n')
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
+ ```