NMS fast mode

This commit is contained in:
Glenn Jocher committed 2020-06-03 13:02:59 -07:00
1 parent d97d31ed6b
commit eb97b2e413
4 files changed
+26 -22

No files matched your search

+1 -1
View File
@@ -76,7 +76,7 @@ def detect(save_img=False):
# Apply NMS
pred = non_max_suppression(pred, opt.conf_thres, opt.iou_thres,
multi_label=False, classes=opt.classes, agnostic=opt.agnostic_nms)
fast=True, classes=opt.classes, agnostic=opt.agnostic_nms)
# Apply Classifier
if classify: