dira/b

2 files changed, 6 insertions, 3 deletions

diff --git a/data/aligned_dataset.py b/data/aligned_dataset.py
index b53a120..bdd6d5f 100755
--- a/data/aligned_dataset.py
+++ b/data/aligned_dataset.py
@@ -19,8 +19,8 @@ class AlignedDataset(BaseDataset):
             dir_B = '_B' if self.opt.label_nc == 0 else '_img'
             self.dir_B = os.path.join(opt.dataroot, opt.phase + 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)))
+            print('train_A length = %d' %len(A_paths))
+            print('train_B length = %d' % len(B_paths))
             self.A_paths = A_paths
             self.B_paths = B_paths
         else:
diff --git a/data/image_folder.py b/data/image_folder.py
index aef0bf5..c286c8f 100755
--- a/data/image_folder.py
+++ b/data/image_folder.py
@@ -35,6 +35,9 @@ def make_aligned_dataset(dir_A, dir_B):
     assert os.path.isdir(dir_A), '%s is not a valid directory' % dir_A
     assert os.path.isdir(dir_B), '%s is not a valid directory' % dir_B
 
+    print(dir_A)
+    print(dir_B)
+
     files_A = [f for f in os.listdir(dir_A) if os.path.isfile(os.path.join(dir_A, f) and is_image_file(f))]
     files_B = [f for f in os.listdir(dir_B) if os.path.isfile(os.path.join(dir_B, f) and is_image_file(f))]
     images = sorted(list(set(files_A).intersection(files_B)))
@@ -42,7 +45,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