From bef5e3cabdc07ecc8920a59652563cf1c30a608e Mon Sep 17 00:00:00 2001 From: Xintao Date: Sun, 8 Aug 2021 15:27:55 +0800 Subject: [PATCH] update for running --- realesrgan/train.py | 8 ++++---- requirements.txt | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) 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