Increase NMS fixed time limit 300ms + 30ms/img (#7957)
Browse files* Increase NMS fixed time limit to 200ms
* Update general.py
- utils/general.py +1 -1
utils/general.py
CHANGED
@@ -795,7 +795,7 @@ def non_max_suppression(prediction,
|
|
795 |
# min_wh = 2 # (pixels) minimum box width and height
|
796 |
max_wh = 7680 # (pixels) maximum box width and height
|
797 |
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
|
798 |
-
time_limit = 0.
|
799 |
redundant = True # require redundant detections
|
800 |
multi_label &= nc > 1 # multiple labels per box (adds 0.5ms/img)
|
801 |
merge = False # use merge-NMS
|
|
|
795 |
# min_wh = 2 # (pixels) minimum box width and height
|
796 |
max_wh = 7680 # (pixels) maximum box width and height
|
797 |
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
|
798 |
+
time_limit = 0.3 + 0.03 * bs # seconds to quit after
|
799 |
redundant = True # require redundant detections
|
800 |
multi_label &= nc > 1 # multiple labels per box (adds 0.5ms/img)
|
801 |
merge = False # use merge-NMS
|