Increase NMS time limit to 50 ms/img (#7956)
Browse files- 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.1 + 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.1 + 0.05 * 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
|