| 123456789101112131415161718192021 |
- # -*- coding: utf-8 -*-
- # @Time : 2020/5/15 17:42
- # @Author : zhoujun
- from addict import Dict
- import copy
- from .RecModel import RecModel
- from .DetModel import DetModel
- from .DistillationDetModel import DistillationModel
- support_model = ['RecModel', 'DetModel','DistillationModel']
- def build_model(config):
- """
- get architecture model class
- """
- copy_config = copy.deepcopy(config)
- arch_type = copy_config.pop('type')
- assert arch_type in support_model, f'{arch_type} is not developed yet!, only {support_model} are support now'
- arch_model = eval(arch_type)(Dict(copy_config))
- return arch_model
|
