first commit

benchmark/PaddleOCR_DBNet/models/head/__init__.py

@@ -0,0 +1,13 @@
+# -*- coding: utf-8 -*-
+# @Time    : 2020/6/5 11:35
+# @Author  : zhoujun
+from .DBHead import DBHead
+
+__all__ = ["build_head"]
+support_head = ["DBHead"]
+
+
+def build_head(head_name, **kwargs):
+    assert head_name in support_head, f"all support head is {support_head}"
+    head = eval(head_name)(**kwargs)
+    return head