Files
insightface/recognition/arcface_oneflow/utils/utils_config.py

19 lines
586 B
Python
Raw Normal View History

2021-10-15 11:30:29 +08:00
import importlib
import os.path as osp
def get_config(config_file):
assert config_file.startswith(
2021-11-04 19:35:06 +08:00
"configs/"
), "config file setting must start with configs/"
2021-10-15 11:30:29 +08:00
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:
2021-11-04 19:35:06 +08:00
cfg.output = osp.join("work_dirs", temp_module_name)
2021-10-15 11:30:29 +08:00
return cfg