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

fix mmedit 2.0 dump info for sdk (#1120)

pull/1091/head
hanrui1sensetime 2022-09-28 16:29:07 +08:00 committed by GitHub
parent f26b352b7d
commit 358c030ad9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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.insert(1, model_cfg.model.data_preprocessor)
preprocess.insert(2, dict(type='ImageToTensor', keys=['img']))
transforms = preprocess
for i, transform in enumerate(transforms):
if 'keys' in transform and transform['keys'] == ['lq']:
transform['keys'] = ['img']
if 'key' in transform and transform['key'] == 'lq':
transform['key'] = 'img'
if transform['type'] == 'ToTensor':
transform['type'] = 'ImageToTensor'
if transform['type'] == 'EditDataPreprocessor':
transform['type'] = 'Normalize'
transform['to_rgb'] = transform.get('to_rgb', False)