mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
09be668a27
* align mmedit static cfg * add docstring for mmcls * update * update for mmdet * update for mmseg * update for mmocr * resolve comments * add doc for mmedit * add docstring for mmcls * update * update for mmdet * update for mmseg * update for mmocr * add doc for mmedit * resolve comments * resolve comments * add type hint * resolve comments * resolve comments Co-authored-by: SingleZombie <singlezombie@163.com>
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from typing import Union
|
|
|
|
import mmcv
|
|
|
|
from mmdeploy.utils import load_config
|
|
|
|
|
|
def dump_info(deploy_cfg: Union[str, mmcv.Config],
|
|
model_cfg: Union[str, mmcv.Config], work_dir: str):
|
|
"""Export information to SDK.
|
|
|
|
Args:
|
|
deploy_cfg (str | mmcv.Config): Deploy config file or dict.
|
|
model_cfg (str | mmcv.Config): Model config file or dict.
|
|
work_dir (str): Work dir to save json files.
|
|
"""
|
|
# TODO dump default values of transformation function to json
|
|
deploy_cfg, model_cfg = load_config(deploy_cfg, model_cfg)
|
|
meta_keys = [
|
|
'filename', 'ori_filename', 'ori_shape', 'img_shape', 'pad_shape',
|
|
'scale_factor', 'flip', 'flip_direction', 'img_norm_cfg'
|
|
]
|
|
if 'transforms' in model_cfg.data.test.pipeline[-1]:
|
|
model_cfg.data.test.pipeline[-1]['transforms'][-1][
|
|
'meta_keys'] = meta_keys
|
|
else:
|
|
model_cfg.data.test.pipeline[-1]['meta_keys'] = meta_keys
|
|
mmcv.dump(
|
|
model_cfg.data.test.pipeline,
|
|
'{}/preprocess.json'.format(work_dir),
|
|
sort_keys=False,
|
|
indent=4)
|
|
|
|
if 'trt' in deploy_cfg:
|
|
deploy_cfg._cfg_dict.pop('trt')
|
|
deploy_cfg.backend_config.common_config.log_level = str(
|
|
deploy_cfg.backend_config.common_config.log_level)
|
|
mmcv.dump(
|
|
deploy_cfg._cfg_dict,
|
|
'{}/deploy_cfg.json'.format(work_dir),
|
|
sort_keys=False,
|
|
indent=4)
|