add unittest for dataset and archs

2021-11-28 15:59:14 +08:00
parent 7dd860a881
commit 1d180efaf3
18 changed files with 227 additions and 2 deletions
--- a/realesrgan/data/realesrgan_paired_dataset.py
+++ b/realesrgan/data/realesrgan_paired_dataset.py
@@ -59,7 +59,7 @@ class RealESRGANPairedDataset(data.Dataset):
            # disk backend with meta_info
            # Each line in the meta_info describes the relative path to an image
            with open(self.opt['meta_info']) as fin:
-                paths = [line.strip().split(' ')[0] for line in fin]
+                paths = [line.strip() for line in fin]
            self.paths = []
            for path in paths:
                gt_path, lq_path = path.split(', ')