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'
|