KaraAgroAI commited on
Commit
30eb2a3
1 Parent(s): 155bc64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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 minimum confidence score that a bounding box must have in order to be kept.
 
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