diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:21:45 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:21:45 +0200 |
| commit | 806da98a2bb49fff816f658d2e72aedf941ba2b6 (patch) | |
| tree | 15063fc5f730cb694eeef65018b16b661a1135f4 /data/image_folder.py | |
| parent | e5f451c6d5156a364a51b43c6ca600dd39c4b04b (diff) | |
dira/b
Diffstat (limited to 'data/image_folder.py')
| -rwxr-xr-x | data/image_folder.py | 5 |
1 files changed, 4 insertions, 1 deletions
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 |