fix version issue

microsoft · Nov 23, 2021 · 2442032 · 2442032
1 parent dd091dc
commit 2442032
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ssod/models/soft_teacher.py b/ssod/models/soft_teacher.py
@@ -164,7 +164,7 @@ def rpn_loss(
                 "rpn_proposal", self.student.test_cfg.rpn
             )
             proposal_list = self.student.rpn_head.get_bboxes(
-                *rpn_out, img_metas, cfg=proposal_cfg
+                *rpn_out, img_metas == img_metas, cfg=proposal_cfg
             )
             log_image_with_boxes(
                 "rpn",
@@ -354,7 +354,7 @@ def extract_teacher_info(self, img, img_metas, proposals=None, **kwargs):
             )
             rpn_out = list(self.teacher.rpn_head(feat))
             proposal_list = self.teacher.rpn_head.get_bboxes(
-                *rpn_out, img_metas, cfg=proposal_cfg
+                *rpn_out, img_metas=img_metas, cfg=proposal_cfg
             )
         else:
             proposal_list = proposals