Browse Source

update for running

Xintao 3 years ago
parent
commit
bef5e3cabd
2 changed files with 5 additions and 5 deletions
  1. 4 4
      realesrgan/train.py
  2. 1 1
      requirements.txt

+ 4 - 4
realesrgan/train.py

@@ -2,10 +2,10 @@
 import os.path as osp
 from basicsr.train import train_pipeline
 
-from .archs import *
-from .data import *
-from .models import *
+import realesrgan.archs
+import realesrgan.data
+import realesrgan.models
 
 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)

+ 1 - 1
requirements.txt

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