From bbd12c7cfef914299db7ced156c6781f999af2b8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 22 Jun 2020 10:14:58 -0700 Subject: [PATCH] detect.py fast bug fix --- detect.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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