From 1b9143d647c9d5322e4face63beaee7f1cd578ca Mon Sep 17 00:00:00 2001 From: THU17cyz Date: Tue, 3 Aug 2021 22:42:45 +0800 Subject: [PATCH 1/2] Fix config --- configs/_base_/models/imvotenet_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/_base_/models/imvotenet_image.py b/configs/_base_/models/imvotenet_image.py index c5067f2c5..37d984f89 100644 --- a/configs/_base_/models/imvotenet_image.py +++ b/configs/_base_/models/imvotenet_image.py @@ -76,7 +76,7 @@ nms_pre=2000, nms_post=1000, max_num=1000, - nms_thr=0.7, + nms=dict(type='nms', iou_threshold=0.7), min_bbox_size=0), img_rcnn=dict( assigner=dict( From 0a74216e70041f58cf36ac7d70e19b31cbb61efa Mon Sep 17 00:00:00 2001 From: THU17cyz Date: Wed, 4 Aug 2021 09:24:05 +0800 Subject: [PATCH 2/2] Fix another place --- configs/_base_/models/imvotenet_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/_base_/models/imvotenet_image.py b/configs/_base_/models/imvotenet_image.py index 37d984f89..981f8bc9b 100644 --- a/configs/_base_/models/imvotenet_image.py +++ b/configs/_base_/models/imvotenet_image.py @@ -75,7 +75,7 @@ nms_across_levels=False, nms_pre=2000, nms_post=1000, - max_num=1000, + max_per_img=1000, nms=dict(type='nms', iou_threshold=0.7), min_bbox_size=0), img_rcnn=dict(