Update NMS max_wh=7680 for 8k images (#6178)

This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-01-03 10:54:52 -08:00
1 parent 5344e54da6
commit b4a29b5a8d
1 file changed
+1 -1
+1 -1
View File
@@ -660,7 +660,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0'
# Settings
min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height
min_wh, max_wh = 2, 7680 # (pixels) minimum and maximum box width and height
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
time_limit = 10.0 # seconds to quit after
redundant = True # require redundant detections