diff options
Diffstat (limited to 'data/aligned_dataset.py')
| -rwxr-xr-x | data/aligned_dataset.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/data/aligned_dataset.py b/data/aligned_dataset.py index bb6947e..b53a120 100755 --- a/data/aligned_dataset.py +++ b/data/aligned_dataset.py @@ -18,7 +18,11 @@ class AlignedDataset(BaseDataset): if opt.isTrain: dir_B = '_B' if self.opt.label_nc == 0 else '_img' self.dir_B = os.path.join(opt.dataroot, opt.phase + dir_B) - self.A_paths, self.B_paths = make_aligned_dataset(self.dir_A, self.dir_B) + A_paths, B_paths = make_aligned_dataset(self.dir_A, self.dir_B) + print('train_A length = %'.format(len(A_paths))) + print('train_B length = %'.format(len(B_paths))) + self.A_paths = A_paths + self.B_paths = B_paths else: self.A_paths = sorted(make_dataset(self.dir_A)) |