diff --git a/configs/_base_/models/imvotenet_image.py b/configs/_base_/models/imvotenet_image.py index c5067f2c5..981f8bc9b 100644 --- a/configs/_base_/models/imvotenet_image.py +++ b/configs/_base_/models/imvotenet_image.py @@ -75,8 +75,8 @@ nms_across_levels=False, nms_pre=2000, nms_post=1000, - max_num=1000, - nms_thr=0.7, + max_per_img=1000, + nms=dict(type='nms', iou_threshold=0.7), min_bbox_size=0), img_rcnn=dict( assigner=dict(