Commit
·
2796768
1
Parent(s):
fb701a7
Update README.md
Browse files
README.md
CHANGED
@@ -92,7 +92,7 @@ To get started with the YOLOv8s object Detection and Classification model, follo
|
|
92 |
pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
|
93 |
```
|
94 |
|
95 |
-
|
96 |
|
97 |
```python
|
98 |
from ultralyticsplus import YOLO, render_result
|
@@ -106,9 +106,8 @@ model.overrides['iou'] = 0.45 # NMS IoU threshold
|
|
106 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
107 |
model.overrides['max_det'] = 1000 # maximum number of detections per image
|
108 |
|
109 |
-
|
110 |
# set image
|
111 |
-
image = 'path/to/your/
|
112 |
|
113 |
# perform inference
|
114 |
results = model.predict(image)
|
@@ -119,6 +118,7 @@ render = render_result(model=model, image=image, result=results[0])
|
|
119 |
render.show()
|
120 |
```
|
121 |
|
|
|
122 |
- Finetune the model on your custom dataset:
|
123 |
|
124 |
```bash
|
|
|
92 |
pip install ultralyticsplus==0.0.28 ultralytics==8.0.43
|
93 |
```
|
94 |
|
95 |
+
- Load model and perform prediction:
|
96 |
|
97 |
```python
|
98 |
from ultralyticsplus import YOLO, render_result
|
|
|
106 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
107 |
model.overrides['max_det'] = 1000 # maximum number of detections per image
|
108 |
|
|
|
109 |
# set image
|
110 |
+
image = '/path/to/your/document/images'
|
111 |
|
112 |
# perform inference
|
113 |
results = model.predict(image)
|
|
|
118 |
render.show()
|
119 |
```
|
120 |
|
121 |
+
|
122 |
- Finetune the model on your custom dataset:
|
123 |
|
124 |
```bash
|