diff --git a/ssod/models/soft_teacher.py b/ssod/models/soft_teacher.py index 380f919..1f53c25 100644 --- 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