detect.py fast bug fix

ultralytics · Jun 22, 2020 · bbd12c7 · bbd12c7
1 parent fc171e2
commit bbd12c7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/detect.py b/detect.py
@@ -64,8 +64,7 @@ def detect(save_img=False):
         pred = model(img, augment=opt.augment)[0]
 
         # Apply NMS
-        pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
-                                   fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
+        pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres, classes=opt.classes, agnostic=opt.agnostic_nms)
         t2 = torch_utils.time_synchronized()
 
         # Apply Classifier