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
|