[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(