|
import os.path as osp |
|
from typing import Dict, List, Union |
|
|
|
from mmengine.config import ConfigDict |
|
|
|
|
|
def model_abbr_from_cfg(cfg: Union[ConfigDict, List[ConfigDict]]) -> str: |
|
"""Generate model abbreviation from the model's confg.""" |
|
if isinstance(cfg, (list, tuple)): |
|
return '_'.join(model_abbr_from_cfg(c) for c in cfg) |
|
if 'abbr' in cfg: |
|
return cfg['abbr'] |
|
model_abbr = cfg['type'] + '_' + '_'.join( |
|
osp.realpath(cfg['path']).split('/')[-2:]) |
|
model_abbr = model_abbr.replace('/', '_') |
|
return model_abbr |
|
|
|
|
|
def dataset_abbr_from_cfg(cfg: ConfigDict) -> str: |
|
"""Returns dataset abbreviation from the dataset's confg.""" |
|
if 'abbr' in cfg: |
|
return cfg['abbr'] |
|
dataset_abbr = cfg['path'] |
|
if 'name' in cfg: |
|
dataset_abbr += '_' + cfg['name'] |
|
dataset_abbr = dataset_abbr.replace('/', '_') |
|
return dataset_abbr |
|
|
|
|
|
def task_abbr_from_cfg(task: Dict) -> str: |
|
"""Returns task abbreviation from the task's confg.""" |
|
return '[' + ','.join([ |
|
f'{model_abbr_from_cfg(model)}/' |
|
f'{dataset_abbr_from_cfg(dataset)}' |
|
for i, model in enumerate(task['models']) |
|
for dataset in task['datasets'][i] |
|
]) + ']' |
|
|
|
|
|
def get_infer_output_path(model_cfg: ConfigDict, |
|
dataset_cfg: ConfigDict, |
|
root_path: str = None, |
|
file_extension: str = 'json') -> str: |
|
|
|
assert root_path is not None, 'default root_path is not allowed any more' |
|
model_abbr = model_abbr_from_cfg(model_cfg) |
|
dataset_abbr = dataset_abbr_from_cfg(dataset_cfg) |
|
return osp.join(root_path, model_abbr, f'{dataset_abbr}.{file_extension}') |
|
|
