| 1234567891011121314151617181920 |
- from .elandarknet.build import build_elandarknet
- from .cspdarknet.build import build_cspdarknet
- from .darknet.build import build_darknet
- from .gelan.build import build_gelan
- def build_model(args):
- # --------------------------- ResNet series ---------------------------
- if 'elandarknet' in args.model:
- model = build_elandarknet(args)
- elif 'cspdarknet' in args.model:
- model = build_cspdarknet(args)
- elif 'darknet' in args.model:
- model = build_darknet(args)
- elif 'gelan' in args.model:
- model = build_gelan(args)
- else:
- raise NotImplementedError("Unknown model: {}".format(args.model))
- return model
|
