diff --git a/mmdet/models/detectors/rpn.py b/mmdet/models/detectors/rpn.py index b882024c97ff01dcdb80a0e5f18f4f4003e82572..947327c22d0846e706ec0359daaee0e5aebc1a11 100644 --- a/mmdet/models/detectors/rpn.py +++ b/mmdet/models/detectors/rpn.py @@ -71,7 +71,7 @@ class RPN(BaseDetector, RPNTestMixin): # TODO: remove this restriction return proposal_list[0].cpu().numpy() - def show_result(self, data, result, img_norm_cfg): + def show_result(self, data, result, img_norm_cfg, dataset=None, top_k=20): """Show RPN proposals on the image. Although we assume batch size is 1, this method supports arbitrary @@ -84,4 +84,4 @@ class RPN(BaseDetector, RPNTestMixin): for img, img_meta in zip(imgs, img_metas): h, w, _ = img_meta['img_shape'] img_show = img[:h, :w, :] - mmcv.imshow_bboxes(img_show, result, top_k=20) + mmcv.imshow_bboxes(img_show, result, top_k=top_k)