Skip to content

Commit b4a29b5

Browse files
authored
Update NMS max_wh=7680 for 8k images (#6178)
1 parent 5344e54 commit b4a29b5

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

utils/general.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
660660
assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0'
661661

662662
# Settings
663-
min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height
663+
min_wh, max_wh = 2, 7680 # (pixels) minimum and maximum box width and height
664664
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
665665
time_limit = 10.0 # seconds to quit after
666666
redundant = True # require redundant detections

0 commit comments

Comments
 (0)