Increase NMS time limit to 50 ms/img (#7956)

2022-05-24 13:58:10 +02:00 · 2022-05-24 13:58:10 +02:00 · b5ebd47d64
parent c215878f11
commit b5ebd47d64
1 changed files with 1 additions and 1 deletions
--- 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