mirror of
https://github.com/PaddlePaddle/PaddleOCR.git
synced 2025-06-03 21:53:39 +08:00
68099c2d5b
* Add custom detection and recognition model usage instructions in re * update * Add custom detection and recognition model usage instructions in re * add db net for benchmark * rename benckmark to PaddleOCR_benchmark * add addict to req * rename
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
|