dira/b

1 files changed, 7 insertions, 7 deletions

diff --git a/data/image_folder.py b/data/image_folder.py
index ba675a3..8fa64f9 100755
--- a/data/image_folder.py
+++ b/data/image_folder.py
@@ -30,18 +30,18 @@ def make_dataset(dir):
 
     return images
 
-def make_aligned_dataset(dir, dir_b):
+def make_aligned_dataset(dir_A, dir_B):
     images = []
-    assert os.path.isdir(dir), '%s is not a valid directory' % dir
-    assert os.path.isdir(dir_b), '%s is not a valid directory' % 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
 
-    files_A = [f for f in os.listdir(dir) if os.path.isfile(os.path.join(mypath, f) and is_image_file(f))]
-    files_B = [f for f in os.listdir(dir_b) if os.path.isfile(os.path.join(mypath, f) and is_image_file(f))]
+    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)))
                 # path = os.path.join(root, fname)
                 # images.append(path)
-    images_A = [os.path.join(dir, f) for f in images]
-    images_B = [os.path.join(dir_b, f) for f in images]
+    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