detect.py fast bug fix
This commit is contained in:
1 parent
fc171e2678
commit
bbd12c7cfe
1 file changed
+1
-2
@@ -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
|
||||
|
||||
Reference in new issue
Block a user