mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Consider compatibility

pull/999/head
whhuang 2022-09-05 12:03:04 +08:00
parent 5830e71c35
commit 93ef8a36b2
1 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
mmdeploy/codebase/mmdet/deploy/object_detection_model.py
View File

@ -281,21 +281,17 @@ class End2EndModel(BaseBackendModel):
win_name: str = '', win_name: str = '',
show: bool = True, show: bool = True,
score_thr: float = 0.3, score_thr: float = 0.3,
bbox_color=(72, 101, 241), out_file=None,
text_color=(72, 101, 241), **kwargs):
mask_color=None,
out_file=None):
return BaseDetector.show_result( return BaseDetector.show_result(
self, self,
img=img, img=img,
result=result, result=result,
score_thr=score_thr, score_thr=score_thr,
bbox_color=bbox_color,
text_color=text_color,
mask_color=mask_color,
show=show, show=show,
win_name=win_name, win_name=win_name,
out_file=out_file) out_file=out_file,
**kwargs)
@__BACKEND_MODEL.register_module('single_stage') @__BACKEND_MODEL.register_module('single_stage')