diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:25:07 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-09-24 15:25:07 +0200 |
| commit | 9934f2a1a5f3356eeab5c3320bd2184decf43c01 (patch) | |
| tree | 52e830895af5c04f2c1195065a74fab669bbf0d6 | |
| parent | 806da98a2bb49fff816f658d2e72aedf941ba2b6 (diff) | |
dira/b
| -rwxr-xr-x | data/image_folder.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/data/image_folder.py b/data/image_folder.py index c286c8f..7f5c6af 100755 --- a/data/image_folder.py +++ b/data/image_folder.py @@ -38,8 +38,8 @@ def make_aligned_dataset(dir_A, 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))] + 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) |