first commit

benchmark/PaddleOCR_DBNet/models/__init__.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+# @Time    : 2019/8/23 21:55
+# @Author  : zhoujun
+import copy
+from .model import Model
+from .losses import build_loss
+
+__all__ = ["build_loss", "build_model"]
+support_model = ["Model"]
+
+
+def build_model(config):
+    """
+    get architecture model class
+    """
+    copy_config = copy.deepcopy(config)
+    arch_type = copy_config.pop("type")
+    assert (
+        arch_type in support_model
+    ), f"{arch_type} is not developed yet!, only {support_model} are support now"
+    arch_model = eval(arch_type)(copy_config)
+    return arch_model