KaraAgroAI
commited on
Commit
•
30eb2a3
1
Parent(s):
155bc64
Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,8 @@ model = torch.hub.load('ultralytics/yolov5', 'KaraAgroAI/CADI-AI')
|
|
39 |
img = ['/path/to/CADI-AI-image.jpg']# batch of images
|
40 |
|
41 |
# set model parameters
|
42 |
-
# set Non-Maximum-Suppression(NMS) threshold to define
|
|
|
43 |
model.conf = 0.20 # NMS confidence threshold
|
44 |
|
45 |
# perform inference
|
|
|
39 |
img = ['/path/to/CADI-AI-image.jpg']# batch of images
|
40 |
|
41 |
# set model parameters
|
42 |
+
# set Non-Maximum-Suppression(NMS) threshold to define
|
43 |
+
# minimum confidence score that a bounding box must have in order to be kept.
|
44 |
model.conf = 0.20 # NMS confidence threshold
|
45 |
|
46 |
# perform inference
|