diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:20:19 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:20:19 +0200 |
| commit | e5f451c6d5156a364a51b43c6ca600dd39c4b04b (patch) | |
| tree | 171c364781893bb294edc02f70b8138beb83fd32 /data/aligned_dataset.py | |
| parent | c2233730d253ddee13a8afd07650e8392774c242 (diff) | |
dira/b
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)) |