[Fix] Fix config of ImVoteNet (#828)

* Fix config * Fix another place
open-mmlab · Aug 4, 2021 · 1ed2b5a · 1ed2b5a
1 parent a876a47
commit 1ed2b5a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(