PaddleOCR/benchmark/PaddleOCR_DBNet/models/neck/__init__.py

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