glenn-jocher
commited on
Commit
•
b4a29b5
1
Parent(s):
5344e54
Update NMS `max_wh=7680` for 8k images (#6178)
Browse files- utils/general.py +1 -1
utils/general.py
CHANGED
@@ -660,7 +660,7 @@ def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=Non
|
|
660 |
assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0'
|
661 |
|
662 |
# Settings
|
663 |
-
min_wh, max_wh = 2,
|
664 |
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
|
665 |
time_limit = 10.0 # seconds to quit after
|
666 |
redundant = True # require redundant detections
|
|
|
660 |
assert 0 <= iou_thres <= 1, f'Invalid IoU {iou_thres}, valid values are between 0.0 and 1.0'
|
661 |
|
662 |
# Settings
|
663 |
+
min_wh, max_wh = 2, 7680 # (pixels) minimum and maximum box width and height
|
664 |
max_nms = 30000 # maximum number of boxes into torchvision.ops.nms()
|
665 |
time_limit = 10.0 # seconds to quit after
|
666 |
redundant = True # require redundant detections
|