dira/b

2 files changed, 6 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))
 
diff --git a/data/image_folder.py b/data/image_folder.py
index 8fa64f9..aef0bf5 100755
--- a/data/image_folder.py
+++ b/data/image_folder.py
@@ -42,6 +42,7 @@ def make_aligned_dataset(dir_A, dir_B):
                 # images.append(path)
     images_A = [os.path.join(dir_A, f) for f in images]
     images_B = [os.path.join(dir_B, f) for f in images]
+    
     return images_A, images_B