glenn-jocher commited on
Commit
bbd12c7
1 Parent(s): fc171e2

detect.py fast bug fix

Browse files
Files changed (1) hide show
  1. detect.py +1 -2
detect.py CHANGED
@@ -64,8 +64,7 @@ def detect(save_img=False):
64
  pred = model(img, augment=opt.augment)[0]
65
 
66
  # Apply NMS
67
- pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
68
- fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
69
  t2 = torch_utils.time_synchronized()
70
 
71
  # Apply Classifier
 
64
  pred = model(img, augment=opt.augment)[0]
65
 
66
  # Apply NMS
67
+ pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
 
68
  t2 = torch_utils.time_synchronized()
69
 
70
  # Apply Classifier