__init__.py 500 B

12345678910
  1. import importlib
  2. from basicsr.utils import scandir
  3. from os import path as osp
  4. # automatically scan and import arch modules for registry
  5. # scan all the files that end with '_arch.py' under the archs folder
  6. arch_folder = osp.dirname(osp.abspath(__file__))
  7. arch_filenames = [osp.splitext(osp.basename(v))[0] for v in scandir(arch_folder) if v.endswith('_arch.py')]
  8. # import all the arch modules
  9. _arch_modules = [importlib.import_module(f'realesrgan.archs.{file_name}') for file_name in arch_filenames]