From 93ef8a36b27b56e8af2e6c71b7f3e0010d0c77da Mon Sep 17 00:00:00 2001 From: whhuang Date: Mon, 5 Sep 2022 12:03:04 +0800 Subject: [PATCH] Consider compatibility --- .../codebase/mmdet/deploy/object_detection_model.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py index e4c555ef2..e89d7af0e 100644 --- a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py +++ b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py @@ -281,21 +281,17 @@ class End2EndModel(BaseBackendModel): win_name: str = '', show: bool = True, score_thr: float = 0.3, - bbox_color=(72, 101, 241), - text_color=(72, 101, 241), - mask_color=None, - out_file=None): + out_file=None, + **kwargs): return BaseDetector.show_result( self, img=img, result=result, score_thr=score_thr, - bbox_color=bbox_color, - text_color=text_color, - mask_color=mask_color, show=show, win_name=win_name, - out_file=out_file) + out_file=out_file, + **kwargs) @__BACKEND_MODEL.register_module('single_stage')