# -*- coding: utf-8 -*- # @Time : 2020/6/5 11:34 # @Author : zhoujun from .FPN import FPN __all__ = ['build_neck'] support_neck = ['FPN'] def build_neck(neck_name, **kwargs): assert neck_name in support_neck, f'all support neck is {support_neck}' neck = eval(neck_name)(**kwargs) return neck