Add files via upload

1 files changed, 29 insertions, 0 deletions

diff --git a/r_basicsr/models/__init__.py b/r_basicsr/models/__init__.py
new file mode 100644
index 0000000..b01cdba
--- /dev/null
+++ b/r_basicsr/models/__init__.py
@@ -0,0 +1,29 @@
+import importlib

+from copy import deepcopy

+from os import path as osp

+

+from r_basicsr.utils import get_root_logger, scandir

+from r_basicsr.utils.registry import MODEL_REGISTRY

+

+__all__ = ['build_model']

+

+# automatically scan and import model modules for registry

+# scan all the files under the 'models' folder and collect files ending with '_model.py'

+model_folder = osp.dirname(osp.abspath(__file__))

+model_filenames = [osp.splitext(osp.basename(v))[0] for v in scandir(model_folder) if v.endswith('_model.py')]

+# import all the model modules

+_model_modules = [importlib.import_module(f'r_basicsr.models.{file_name}') for file_name in model_filenames]

+

+

+def build_model(opt):

+    """Build model from options.

+

+    Args:

+        opt (dict): Configuration. It must contain:

+            model_type (str): Model type.

+    """

+    opt = deepcopy(opt)

+    model = MODEL_REGISTRY.get(opt['model_type'])(opt)

+    logger = get_root_logger()

+    logger.info(f'Model [{model.__class__.__name__}] is created.')

+    return model