first commit

benchmark/PaddleOCR_DBNet/models/neck/__init__.py

@@ -0,0 +1,13 @@
+# -*- 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