From e5f451c6d5156a364a51b43c6ca600dd39c4b04b Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 24 Sep 2018 15:20:19 +0200 Subject: dira/b --- data/aligned_dataset.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'data/aligned_dataset.py') 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)) -- cgit v1.2.3-70-g09d2