Update README.md
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ model-index:
|
|
| 40 |
- Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus):
|
| 41 |
|
| 42 |
```bash
|
| 43 |
-
pip install ultralyticsplus==0.0.28
|
| 44 |
```
|
| 45 |
|
| 46 |
- Load model and perform prediction:
|
|
@@ -52,10 +52,10 @@ from ultralyticsplus import YOLO, render_result
|
|
| 52 |
model = YOLO('eeshawn11/naruto_hand_seal_detection')
|
| 53 |
|
| 54 |
# set model parameters
|
| 55 |
-
model.overrides['conf'] = 0.
|
| 56 |
-
model.overrides['iou'] = 0.
|
| 57 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
| 58 |
-
model.overrides['max_det'] =
|
| 59 |
|
| 60 |
# set image
|
| 61 |
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
|
|
|
|
| 40 |
- Install [ultralyticsplus](https://github.com/fcakyon/ultralyticsplus):
|
| 41 |
|
| 42 |
```bash
|
| 43 |
+
pip install ultralyticsplus==0.0.28
|
| 44 |
```
|
| 45 |
|
| 46 |
- Load model and perform prediction:
|
|
|
|
| 52 |
model = YOLO('eeshawn11/naruto_hand_seal_detection')
|
| 53 |
|
| 54 |
# set model parameters
|
| 55 |
+
model.overrides['conf'] = 0.50 # NMS confidence threshold
|
| 56 |
+
model.overrides['iou'] = 0.70 # NMS IoU threshold
|
| 57 |
model.overrides['agnostic_nms'] = False # NMS class-agnostic
|
| 58 |
+
model.overrides['max_det'] = 10 # maximum number of detections per image
|
| 59 |
|
| 60 |
# set image
|
| 61 |
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'
|