diff --git a/detect.py b/detect.py index bc4ce62b7355..567c9e6e7f2d 100644 --- 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