lucasrufkahr/Real-ESRGAN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

regroup

Browse Source
This commit is contained in:
Xintao
2021-08-08 14:50:32 +08:00
parent 9baa0b3d00
commit 94ae626008
2 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
realesrgan/train.py
View File

@@ -1,9 +1,10 @@
# flake8: noqa
import os.path as osp import os.path as osp
from basicsr.train import train_pipeline from basicsr.train import train_pipeline
import realesrgan.archs # noqa: F401 from .archs import *
import realesrgan.data # noqa: F401 from .data import *
import realesrgan.models # noqa: F401 from .models import *
if __name__ == '__main__': if __name__ == '__main__':
root_path = osp.abspath(osp.join(__file__, osp.pardir)) root_path = osp.abspath(osp.join(__file__, osp.pardir))

2
setup.cfg
View File

@@ -16,7 +16,7 @@ split_before_expression_after_opening_paren = true
line_length = 120 line_length = 120
multi_line_output = 0 multi_line_output = 0
known_standard_library = pkg_resources,setuptools known_standard_library = pkg_resources,setuptools
known_first_party = basicsr # modify it! known_first_party = realesrgan
known_third_party = basicsr,cv2,numpy,torch known_third_party = basicsr,cv2,numpy,torch
no_lines_before = STDLIB,LOCALFOLDER no_lines_before = STDLIB,LOCALFOLDER
default_section = THIRDPARTY default_section = THIRDPARTY