# ----------------------- Model Config -----------------------
from .retinanet_config import retinanet_cfg
from .fcos_config      import fcos_cfg
from .yolof_config     import yolof_cfg
from .detr_config      import detr_cfg

def build_config(args):
    # RetinaNet
    if args.model in retinanet_cfg.keys():
        return retinanet_cfg[args.model]
    # FCOS
    elif args.model in fcos_cfg.keys():
        return fcos_cfg[args.model]
    # YOLOF
    elif args.model in yolof_cfg.keys():
        return yolof_cfg[args.model]
    # DETR
    elif args.model in detr_cfg.keys():
        return detr_cfg[args.model]
    else:
        raise NotImplementedError('Unknown Model: {}'.format(args.model))