Windows picture duplicates fix by arnoegw

for #154 PiperOrigin-RevId: 213233024
tensorflow · Sep 17, 2018 · 1fa48fe · 1fa48fe
1 parent 022eb2a
commit 1fa48fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/image_retraining/retrain.py b/examples/image_retraining/retrain.py
@@ -173,7 +173,8 @@ def create_image_lists(image_dir, testing_percentage, validation_percentage):
     if is_root_dir:
       is_root_dir = False
       continue
-    extensions = ['jpg', 'jpeg', 'JPG', 'JPEG']
+    extensions = sorted(set(os.path.normcase(ext)  # Smash case on Windows.
+                            for ext in ['JPEG', 'JPG', 'jpeg', 'jpg']))
     file_list = []
     dir_name = os.path.basename(sub_dir)
     if dir_name == image_dir: