HariLogicgo commited on
Commit
138a9cd
·
1 Parent(s): fbb600e

file calling init.py corrected

Browse files
CodeFormer/basicsr/utils/__init__.py CHANGED
@@ -1,29 +1,11 @@
1
- from .file_client import FileClient
2
- from .img_util import crop_border, imfrombytes, img2tensor, imwrite, tensor2img
3
- from .logger import MessageLogger, get_env_info, get_root_logger, init_tb_logger, init_wandb_logger
4
- from .misc import check_resume, get_time_str, make_exp_dirs, mkdir_and_rename, scandir, set_random_seed, sizeof_fmt
5
-
6
- __all__ = [
7
- # file_client.py
8
- 'FileClient',
9
- # img_util.py
10
- 'img2tensor',
11
- 'tensor2img',
12
- 'imfrombytes',
13
- 'imwrite',
14
- 'crop_border',
15
- # logger.py
16
- 'MessageLogger',
17
- 'init_tb_logger',
18
- 'init_wandb_logger',
19
- 'get_root_logger',
20
- 'get_env_info',
21
- # misc.py
22
- 'set_random_seed',
23
- 'get_time_str',
24
- 'mkdir_and_rename',
25
- 'make_exp_dirs',
26
- 'scandir',
27
- 'check_resume',
28
- 'sizeof_fmt'
29
- ]
 
1
+ # https://github.com/xinntao/BasicSR
2
+ # flake8: noqa
3
+ from archs import *
4
+ from data import *
5
+ from losses import *
6
+ from metrics import *
7
+ from models import *
8
+ from ops import *
9
+ from train import *
10
+ from utils import *
11
+ from version import __gitsha__, __version__
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
CodeFormer/inference_codeformer.py CHANGED
@@ -4,7 +4,7 @@ import argparse
4
  import glob
5
  import torch
6
  from torchvision.transforms.functional import normalize
7
- from basicsr.utils import imwrite, img2tensor, tensor2img
8
  from basicsr.utils.download_util import load_file_from_url
9
  import torch
10
 
 
4
  import glob
5
  import torch
6
  from torchvision.transforms.functional import normalize
7
+ from basicsr.utils.img_util import imwrite, img2tensor, tensor2img
8
  from basicsr.utils.download_util import load_file_from_url
9
  import torch
10