Files
insightface/recognition/arcface_oneflow/utils/utils_config.py
2021-11-04 19:35:06 +08:00

19 lines
586 B
Python

import importlib
import os.path as osp
def get_config(config_file):
assert config_file.startswith(
"configs/"
), "config file setting must start with configs/"
temp_config_name = osp.basename(config_file)
temp_module_name = osp.splitext(temp_config_name)[0]
config = importlib.import_module("configs.base")
cfg = config.config
config = importlib.import_module("configs.%s" % temp_module_name)
job_cfg = config.config
cfg.update(job_cfg)
if cfg.output is None:
cfg.output = osp.join("work_dirs", temp_module_name)
return cfg