Fix pix2pix preprocessing

soochan-lee · May 29, 2019 · 1cbeb75 · 1cbeb75
1 parent 7d976dc
commit 1cbeb75
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -45,8 +45,7 @@ $ python ./scripts/preprocess_pix2pix_data.py \
     --src data/cityscapes/original/leftImg8bit \
     --dst data/cityscapes/256x256 \
     --size 256 \
-    --random-flip \
-    --random-rotate
+    --random-flip
 ```
 
 ### Maps

diff --git a/scripts/preprocess_pix2pix_data.py b/scripts/preprocess_pix2pix_data.py
@@ -24,10 +24,10 @@ def main():
         src_dir = os.path.join(args.src, mode)
         dst_dir = os.path.join(args.dst, mode)
 
-        items = [
-            os.path.join(src_dir, name)
-            for name in os.listdir(src_dir)
-        ]
+        items = []
+        for path, dir, files in os.walk(src_dir):
+            for file in files:
+                items.append(os.path.join(path, file))
 
         index = 0
         print('Processing %s...' % mode)