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

update for running

Browse Source
This commit is contained in:
Xintao
2021-08-08 15:27:55 +08:00
parent 064df9956b
commit bef5e3cabd
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
realesrgan/train.py
View File

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

2
requirements.txt
View File

@@ -1,4 +1,4 @@
basicsr basicsr
cv2
numpy numpy
opencv-python
torch>=1.7 torch>=1.7