mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

fix with_argmax for mmseg (#2056)

Browse Source
This commit is contained in:
RunningLeon 2023-05-11 15:37:53 +08:00 committed by GitHub
parent 9385eed16f
commit 26b66ef511
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmdeploy/codebase/mmseg/deploy/segmentation.py
View File

@ -303,10 +303,10 @@ class Segmentation(BaseTask):
params = self.model_cfg.model.decode_head params = self.model_cfg.model.decode_head
if isinstance(params, list): if isinstance(params, list):
params = params[-1] params = params[-1]
postprocess = dict(params=params, type='ResizeMask')
with_argmax = get_codebase_config(self.deploy_cfg).get( with_argmax = get_codebase_config(self.deploy_cfg).get(
'with_argmax', True) 'with_argmax', True)
postprocess['with_argmax'] = with_argmax params['with_argmax'] = with_argmax
postprocess = dict(params=params, type='ResizeMask')
return postprocess return postprocess
def get_model_name(self, *args, **kwargs) -> str: def get_model_name(self, *args, **kwargs) -> str: