glenn-jocher commited on
Commit
b5ebd47
·
unverified ·
1 Parent(s): c215878

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

Browse files
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.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
 
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