Commit
·
2d56af1
1
Parent(s):
10a6e2e
Update README.md
Browse files
README.md
CHANGED
|
@@ -42,7 +42,7 @@ import torch
|
|
| 42 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='CADI-AI/yolov5_0.65map_exp7_best.pt', force_reload=True)
|
| 43 |
|
| 44 |
# Images
|
| 45 |
-
img = ['/path/to/
|
| 46 |
|
| 47 |
# set model parameters
|
| 48 |
# set Non-Maximum-Suppression(NMS) threshold to define
|
|
|
|
| 42 |
model = torch.hub.load('ultralytics/yolov5', 'custom', path='CADI-AI/yolov5_0.65map_exp7_best.pt', force_reload=True)
|
| 43 |
|
| 44 |
# Images
|
| 45 |
+
img = ['/path/to/your/image.jpg']# batch of images
|
| 46 |
|
| 47 |
# set model parameters
|
| 48 |
# set Non-Maximum-Suppression(NMS) threshold to define
|