jameslahm commited on
Commit
cf2b156
1 Parent(s): 3a8ff09

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -4
README.md CHANGED
@@ -6,6 +6,7 @@ tags:
6
  - yolov10
7
  datasets:
8
  - detection-datasets/coco
 
9
  ---
10
 
11
  ### Model Description
@@ -16,10 +17,10 @@ datasets:
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
 
@@ -31,10 +32,22 @@ 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
  ```
 
6
  - yolov10
7
  datasets:
8
  - detection-datasets/coco
9
+ inference: false
10
  ---
11
 
12
  ### Model Description
 
17
 
18
  ### Installation
19
  ```
20
+ pip install git+https://github.com/THU-MIG/yolov10.git
21
  ```
22
 
23
+ ### Training and validation
24
  ```python
25
  from ultralytics import YOLOv10
26
 
 
32
 
33
  # Validation
34
  model.val(...)
35
+ ```
36
+
37
+ ### Inference
38
+
39
+ Here's an end-to-end example showcasing inference on a cats image:
40
+
41
+ ```python
42
+ from ultralytics import YOLOv10
43
 
44
+ model = YOLOv10.from_pretrained('jameslahm/yolov10b')
45
+ source = 'http://images.cocodataset.org/val2017/000000039769.jpg'
46
+ model.predict(source=source, save=True)
47
  ```
48
+ which shows:
49
+
50
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/628ece6054698ce61d1e7be3/E24iLZLtBXva9uXXgrYmo.png)
51
 
52
  ### BibTeX Entry and Citation Info
53
  ```