mmdet showresult add *args

open-mmlab · Sep 5, 2022 · 82265a3 · 82265a3
1 parent 93ef8a3
commit 82265a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmdeploy/codebase/mmdet/deploy/object_detection_model.py b/mmdeploy/codebase/mmdet/deploy/object_detection_model.py
@@ -276,6 +276,7 @@ def forward_test(self, imgs: torch.Tensor, *args, **kwargs) -> \
         return outputs
 
     def show_result(self,
+                    *args,
                     img: np.ndarray,
                     result: list,
                     win_name: str = '',
@@ -285,6 +286,7 @@ def show_result(self,
                     **kwargs):
         return BaseDetector.show_result(
             self,
+            *args,
             img=img,
             result=result,
             score_thr=score_thr,