Spaces:
Running
Running
File size: 2,298 Bytes
2ba4412 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import os, yaml
from copy import deepcopy, copy
# def get prior and ldm config
def assign_prior_mudule_cfg(cfg):
'''
'''
#
prior_cfg = deepcopy(cfg)
vldm_cfg = deepcopy(cfg)
with open(cfg.prior_cfg, 'r') as f:
_cfg_update = yaml.load(f.read(), Loader=yaml.SafeLoader)
# _cfg_update = _cfg_update.cfg_dict
for k, v in _cfg_update.items():
if isinstance(v, dict) and k in cfg:
prior_cfg[k].update(v)
else:
prior_cfg[k] = v
with open(cfg.vldm_cfg, 'r') as f:
_cfg_update = yaml.load(f.read(), Loader=yaml.SafeLoader)
# _cfg_update = _cfg_update.cfg_dict
for k, v in _cfg_update.items():
if isinstance(v, dict) and k in cfg:
vldm_cfg[k].update(v)
else:
vldm_cfg[k] = v
return prior_cfg, vldm_cfg
# def get prior and ldm config
def assign_vldm_vsr_mudule_cfg(cfg):
'''
'''
#
vldm_cfg = deepcopy(cfg)
vsr_cfg = deepcopy(cfg)
with open(cfg.vldm_cfg, 'r') as f:
_cfg_update = yaml.load(f.read(), Loader=yaml.SafeLoader)
# _cfg_update = _cfg_update.cfg_dict
for k, v in _cfg_update.items():
if isinstance(v, dict) and k in cfg:
vldm_cfg[k].update(v)
else:
vldm_cfg[k] = v
with open(cfg.vsr_cfg, 'r') as f:
_cfg_update = yaml.load(f.read(), Loader=yaml.SafeLoader)
# _cfg_update = _cfg_update.cfg_dict
for k, v in _cfg_update.items():
if isinstance(v, dict) and k in cfg:
vsr_cfg[k].update(v)
else:
vsr_cfg[k] = v
return vldm_cfg, vsr_cfg
# def get prior and ldm config
def assign_signle_cfg(cfg, _cfg_update, tname):
'''
'''
#
vldm_cfg = deepcopy(cfg)
if os.path.exists(_cfg_update[tname]):
with open(_cfg_update[tname], 'r') as f:
_cfg_update = yaml.load(f.read(), Loader=yaml.SafeLoader)
# _cfg_update = _cfg_update.cfg_dict
for k, v in _cfg_update.items():
if isinstance(v, dict) and k in cfg:
vldm_cfg[k].update(v)
else:
vldm_cfg[k] = v
return vldm_cfg |