glenn-jocher commited on
Commit
68ff6c9
1 Parent(s): b5ebd47

Increase NMS fixed time limit 300ms + 30ms/img (#7957)

Browse files

* Increase NMS fixed time limit to 200ms

* Update general.py

Files changed (1) hide show
  1. 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.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
 
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