From c2233730d253ddee13a8afd07650e8392774c242 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 24 Sep 2018 15:17:23 +0200 Subject: dira/b --- data/image_folder.py | 14 +++++++------- 1 file 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 -- cgit v1.2.3-70-g09d2