| from types import ModuleType |
| import data_info |
|
|
| def load_data_info(module_name, data_info={}, mldb_type='mldb_info', module=None): |
| if module is None: |
| module = globals().get(module_name, None) |
| if module: |
| for key, value in module.__dict__.items(): |
| if not (key.startswith('__')) and not (key.startswith('_')): |
| if key == 'mldb_info': |
| data_info.update(value) |
| elif isinstance(value, ModuleType): |
| load_data_info(module_name + '.' + key, data_info, module=value) |
| else: |
| raise RuntimeError(f'Try to access "mldb_info", but cannot find {module_name} module.') |
|
|
| def reset_ckpt_path(cfg, data_info): |
| if isinstance(cfg, dict): |
| for key in cfg.keys(): |
| if key == 'backbone': |
| new_ckpt_path = data_info['checkpoint']['mldb_root'] + '/' + data_info['checkpoint'][cfg.backbone.type] |
| cfg.backbone.update(checkpoint=new_ckpt_path) |
| continue |
| elif isinstance(cfg.get(key), dict): |
| reset_ckpt_path(cfg.get(key), data_info) |
| else: |
| continue |
| else: |
| return |
|
|
| if __name__ == '__main__': |
| mldb_info_tmp = {} |
| load_data_info('mldb_data_info', mldb_info_tmp) |
| print('results', mldb_info_tmp.keys()) |
