Use scandir in _find_classes (#557) (#559)

* Use scandir in _find_classes (#557)

* fix lint

Author: David Wade

torchvision/datasets/folder.py

@@ -4,6 +4,7 @@
 
 import os
 import os.path
+import sys
 
 
 def has_file_allowed_extension(filename, extensions):
@@ -110,7 +111,11 @@ def _find_classes(self, dir):
         Ensures:
             No class is a subdirectory of another.
         """
-        classes = [d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d))]
+        if sys.version_info >= (3, 5):
+            # Faster and available in Python 3.5 and above
+            classes = [d.name for d in os.scandir(dir) if d.is_dir()]
+        else:
+            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