update fast mode

This commit is contained in:
Glenn Jocher committed 2020-06-16 00:11:29 -07:00
1 parent db2c3acd3a
commit 1c0b6236e3
2 files changed
+2 -3

No files matched your search

+1 -2
View File
@@ -526,12 +526,11 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, fast=False, c
time_limit = 10.0 # seconds to quit after
redundant = True # require redundant detections
fast |= conf_thres > 0.001 # fast mode
multi_label = nc > 1 # multiple labels per box (adds 0.5ms/img)
if fast:
merge = False
multi_label = nc > 1 # multiple labels per box (adds 0.5ms/img)
else:
merge = True # merge for best mAP (adds 0.5ms/img)
multi_label = nc > 1 # multiple labels per box (adds 0.5ms/img)
t = time.time()
output = [None] * prediction.shape[0]