YOLO-Tutorial-v2/odlab/config/__init__.py

# ----------------------- Model Config -----------------------
from .fcos_config      import build_fcos_config
from .yolof_config     import build_yolof_config
from .detr_config      import build_detr_config

def build_config(args):
    # FCOS
    if "fcos" in args.model:
        cfg = build_fcos_config(args)
    # YOLOF
    elif "yolof" in args.model:
        cfg = build_yolof_config(args)
    # DETR
    elif "detr" in args.model:
        cfg = build_detr_config(args)
    else:
        raise NotImplementedError('Unknown Model: {}'.format(args.model))

    # Print model config
    cfg.print_config()

    return cfg