init.py corrected

CodeFormer/basicsr/__init__.py

@@ -9,3 +9,4 @@ from .ops import *
 from .train import *
 from .utils import *
 #from .version import __gitsha__, __version__
+

CodeFormer/basicsr/utils/__init__.py

@@ -1,11 +1,29 @@
-# https://github.com/xinntao/BasicSR
-# flake8: noqa
-from archs import *
-from data import *
-from losses import *
-from metrics import *
-from models import *
-from ops import *
-from train import *
-from utils import *
-from version import __gitsha__, __version__
+from .file_client import FileClient
+from .img_util import crop_border, imfrombytes, img2tensor, imwrite, tensor2img
+from .logger import MessageLogger, get_env_info, get_root_logger, init_tb_logger, init_wandb_logger
+from .misc import check_resume, get_time_str, make_exp_dirs, mkdir_and_rename, scandir, set_random_seed, sizeof_fmt
+
+__all__ = [
+    # file_client.py
+    'FileClient',
+    # img_util.py
+    'img2tensor',
+    'tensor2img',
+    'imfrombytes',
+    'imwrite',
+    'crop_border',
+    # logger.py
+    'MessageLogger',
+    'init_tb_logger',
+    'init_wandb_logger',
+    'get_root_logger',
+    'get_env_info',
+    # misc.py
+    'set_random_seed',
+    'get_time_str',
+    'mkdir_and_rename',
+    'make_exp_dirs',
+    'scandir',
+    'check_resume',
+    'sizeof_fmt'
+]

CodeFormer/inference_codeformer.py

@@ -4,7 +4,7 @@ import argparse
 import glob
 import torch
 from torchvision.transforms.functional import normalize
-from basicsr.utils.img_util import imwrite, img2tensor, tensor2img
+from basicsr.utils import imwrite, img2tensor, tensor2img
 from basicsr.utils.download_util import load_file_from_url
 import torch