From b5ebd47d644f5f0d0d30b002fe6f49166f9fea83 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 24 May 2022 13:58:10 +0200 Subject: [PATCH] Increase NMS time limit to 50 ms/img (#7956) --- utils/general.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/general.py b/utils/general.py index ed5cc38..d44898b 100755 --- a/utils/general.py +++ b/utils/general.py @@ -795,7 +795,7 @@ def non_max_suppression(prediction, # min_wh = 2 # (pixels) minimum box width and height max_wh = 7680 # (pixels) maximum box width and height max_nms = 30000 # maximum number of boxes into torchvision.ops.nms() - time_limit = 0.1 + 0.03 * bs # seconds to quit after + time_limit = 0.1 + 0.05 * bs # seconds to quit after redundant = True # require redundant detections multi_label &= nc > 1 # multiple labels per box (adds 0.5ms/img) merge = False # use merge-NMS