|
|
@@ -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 |