28 lines
985 B
Python
28 lines
985 B
Python
#!/usr/bin/env python
|
|
# -*- encoding: utf-8 -*-
|
|
'''
|
|
@File : __init__.py.py
|
|
@Contact : zpyovo@hotmail.com
|
|
@License : (C)Copyright 2018-2019, Lab501-TransferLearning-SCUT
|
|
@Description :
|
|
|
|
@Modify Time @Author @Version @Desciption
|
|
------------ ------- -------- -----------
|
|
2023/8/2 15:55 Pengyu Zhang 1.0 None
|
|
'''
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
import importlib
|
|
from basicsr.utils import scandir
|
|
from os import path as osp
|
|
|
|
# automatically scan and import arch modules for registry
|
|
# scan all the files that end with '_arch.py' under the archs folder
|
|
arch_folder = osp.dirname(osp.abspath(__file__))
|
|
arch_filenames = [osp.splitext(osp.basename(v))[0] for v in scandir(arch_folder) if v.endswith('_modles.py')]
|
|
# import all the arch modules
|
|
_arch_modules = [importlib.import_module(f'yolo5x_qr.models.{file_name}') for file_name in arch_filenames]
|
|
|