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 mmedit 2.0 dump info for sdk (#1120)

Browse Source
This commit is contained in:
hanrui1sensetime 2022-09-28 16:29:07 +08:00 committed by RunningLeon
parent c0ae726ae5
commit b8c19b35d2
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmdeploy/codebase/mmedit/deploy/super_resolution.py
View File

@ -309,14 +309,13 @@ class SuperResolution(BaseTask):
preprocess = model_cfg.test_pipeline preprocess = model_cfg.test_pipeline
preprocess.insert(1, model_cfg.model.data_preprocessor) preprocess.insert(1, model_cfg.model.data_preprocessor)
preprocess.insert(2, dict(type='ImageToTensor', keys=['img']))
transforms = preprocess transforms = preprocess
for i, transform in enumerate(transforms): for i, transform in enumerate(transforms):
if 'keys' in transform and transform['keys'] == ['lq']: if 'keys' in transform and transform['keys'] == ['lq']:
transform['keys'] = ['img'] transform['keys'] = ['img']
if 'key' in transform and transform['key'] == 'lq': if 'key' in transform and transform['key'] == 'lq':
transform['key'] = 'img' transform['key'] = 'img'
if transform['type'] == 'ToTensor':
transform['type'] = 'ImageToTensor'
if transform['type'] == 'EditDataPreprocessor': if transform['type'] == 'EditDataPreprocessor':
transform['type'] = 'Normalize' transform['type'] = 'Normalize'
transform['to_rgb'] = transform.get('to_rgb', False) transform['to_rgb'] = transform.get('to_rgb', False)