[Fix] Fix config of ImVoteNet (#828)

* Fix config * Fix another place

[Fix] Fix config of ImVoteNet (#828)
* Fix config * Fix another place
1ed2b5ab · Yezhen Cong · GitHub · a876a472 · 1ed2b5ab
Unverified Commit 1ed2b5ab authored Aug 04, 2021 by Yezhen Cong Committed by GitHub Aug 04, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

configs/_base_/models/imvotenet_image.py configs/_base_/models/imvotenet_image.py +2 -2

No files found.
--- a/configs/_base_/models/imvotenet_image.py
+++ b/configs/_base_/models/imvotenet_image.py
@@ -75,8 +75,8 @@ model = dict(
            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(