14 lines
311 B
Python
14 lines
311 B
Python
|
# -*- 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
|