From f70164d2c65d15bda8b950804093d6a9d5b2949f Mon Sep 17 00:00:00 2001 From: RangiLyu Date: Fri, 28 Oct 2022 10:14:18 +0800 Subject: [PATCH] [Fix] Fix RPN visualization (#9151) * [Fix] Fix RPN visualization * [Fix] Fix RPN visualization --- mmdet/models/detectors/rpn.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/mmdet/models/detectors/rpn.py b/mmdet/models/detectors/rpn.py index 6ec326b75..707e02b0e 100644 --- a/mmdet/models/detectors/rpn.py +++ b/mmdet/models/detectors/rpn.py @@ -1,5 +1,6 @@ # Copyright (c) OpenMMLab. All rights reserved. import warnings +from inspect import signature import mmcv import torch @@ -153,7 +154,9 @@ class RPN(BaseDetector): np.ndarray: The image with bboxes drawn on it. """ if kwargs is not None: - kwargs.pop('score_thr', None) - kwargs.pop('text_color', None) - kwargs['colors'] = kwargs.pop('bbox_color', 'green') + kwargs['colors'] = 'green' + sig = signature(mmcv.imshow_bboxes) + for k in list(kwargs.keys()): + if k not in sig.parameters: + kwargs.pop(k) mmcv.imshow_bboxes(data, result, top_k=top_k, **kwargs)