From 68ff6c9c08dd65feb7b7127af23c6900b3906ec2 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 24 May 2022 14:11:03 +0200
Subject: [PATCH] Increase NMS fixed time limit 300ms + 30ms/img (#7957)

* Increase NMS fixed time limit to 200ms

* Update general.py
---
 utils/general.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/general.py b/utils/general.py
index d44898b..27c439e 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.05 * bs  # seconds to quit after
+    time_limit = 0.3 + 0.03 * 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