minhah commited on
Commit
64d7585
1 Parent(s): 61c3b9b

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +62 -0
  2. config.json +7 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - detection-datasets/coco
4
+ library_name: ultralytics
5
+ license: agpl-3.0
6
+ tags:
7
+ - object-detection
8
+ - computer-vision
9
+ - yolov10
10
+ repo_url: https://github.com/THU-MIG/yolov10
11
+ inference: false
12
+ ---
13
+
14
+ ### Model Description
15
+ [YOLOv10: Real-Time End-to-End Object Detection](https://arxiv.org/abs/2405.14458v1)
16
+
17
+ - arXiv: https://arxiv.org/abs/2405.14458v1
18
+ - github: https://github.com/THU-MIG/yolov10
19
+
20
+ ### Installation
21
+ ```
22
+ pip install git+https://github.com/THU-MIG/yolov10.git
23
+ ```
24
+
25
+ ### Training and validation
26
+ ```python
27
+ from ultralytics import YOLOv10
28
+
29
+ model = YOLOv10.from_pretrained('jameslahm/yolov10n')
30
+ # Training
31
+ model.train(...)
32
+ # after training, one can push to the hub
33
+ model.push_to_hub("your-hf-username/yolov10-finetuned")
34
+
35
+ # Validation
36
+ model.val(...)
37
+ ```
38
+
39
+ ### Inference
40
+
41
+ Here's an end-to-end example showcasing inference on a cats image:
42
+
43
+ ```python
44
+ from ultralytics import YOLOv10
45
+
46
+ model = YOLOv10.from_pretrained('jameslahm/yolov10n')
47
+ source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
48
+ model.predict(source=source, save=True)
49
+ ```
50
+ which shows:
51
+
52
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/628ece6054698ce61d1e7be3/tBwAsKcQA_96HCYQp7BRr.png)
53
+
54
+ ### BibTeX Entry and Citation Info
55
+ ```
56
+ @article{wang2024yolov10,
57
+ title={YOLOv10: Real-Time End-to-End Object Detection},
58
+ author={Wang, Ao and Chen, Hui and Liu, Lihao and Chen, Kai and Lin, Zijia and Han, Jungong and Ding, Guiguang},
59
+ journal={arXiv preprint arXiv:2405.14458},
60
+ year={2024}
61
+ }
62
+ ```
config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "yolov10s.yaml",
3
+ "names": {
4
+ "0": ""
5
+ },
6
+ "task": "detect"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a562466ab5aa6bc00ab08c0e5313136e33ff98c2dc0e454e2cd58e77862815f
3
+ size 32502160