# -*- 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