diff --git a/realesrgan/train.py b/realesrgan/train.py
index e9ad4dd..8a9cec9 100644
--- a/realesrgan/train.py
+++ b/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)
diff --git a/requirements.txt b/requirements.txt
index 0ea6378..f0a64cb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 basicsr
-cv2
 numpy
+opencv-python
 torch>=1.7