vanillo commited on
Commit
048eac7
1 Parent(s): e679ccf

Push model using huggingface_hub.

Browse files
Files changed (3) hide show
  1. README.md +62 -0
  2. config.json +12 -0
  3. model.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: agpl-3.0
3
+ library_name: ultralytics
4
+ tags:
5
+ - object-detection
6
+ - computer-vision
7
+ - yolov10
8
+ datasets:
9
+ - detection-datasets/coco
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,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "yolov10n.yaml",
3
+ "names": {
4
+ "0": "fire",
5
+ "1": "light",
6
+ "2": "smoke",
7
+ "3": "reflect",
8
+ "4": "face",
9
+ "5": "like-smoke"
10
+ },
11
+ "task": "detect"
12
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7433aedddb061a5a071adf32e0223727bf81b16c9256e92bf017318409b66b16
3
+ size 10978912