From 80a223da184fac701c14123bb4ccf043c12a5621 Mon Sep 17 00:00:00 2001
From: Kai Chen <chenkaidev@gmail.com>
Date: Thu, 14 Feb 2019 22:24:02 +0800
Subject: [PATCH] bug fix for rpn proposal visualization
---
mmdet/models/detectors/rpn.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mmdet/models/detectors/rpn.py b/mmdet/models/detectors/rpn.py
index b882024..947327c 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)
--
GitLab