DS_Store fix in find_classes (#79)

c0a6cfe1 · NC Cullen · Soumith Chintala · 39a50de1 · c0a6cfe1
Commit c0a6cfe1 authored Feb 27, 2017 by NC Cullen Committed by Soumith Chintala Feb 27, 2017
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/folder.py torchvision/datasets/folder.py +1 -1

No files found.
--- a/torchvision/datasets/folder.py
+++ b/torchvision/datasets/folder.py
@@ -15,7 +15,7 @@ def is_image_file(filename):


 def find_classes(dir):
-    classes = os.listdir(dir)
+    classes = [d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d))]
    classes.sort()
    class_to_idx = {classes[i]: i for i in range(len(classes))}
    return classes, class_to_idx