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
|