| 1234567891011121314151617 |
- # ----------------------- Model Config -----------------------
- from .fcos_config import fcos_cfg
- from .yolof_config import yolof_cfg
- from .detr_config import detr_cfg
- def build_config(args):
- # FCOS
- if 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))
|
