Faster dataset indexing (#3939)

9b29f3f2 · IgorSusmelj · GitHub · 234d262f · 9b29f3f2
Unverified Commit 9b29f3f2 authored Jun 01, 2021 by IgorSusmelj Committed by GitHub Jun 01, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

No files found.
--- a/torchvision/datasets/folder.py
+++ b/torchvision/datasets/folder.py
@@ -86,8 +86,8 @@ def make_dataset(
            continue
        for root, _, fnames in sorted(os.walk(target_dir, followlinks=True)):
            for fname in sorted(fnames):
-                path = os.path.join(root, fname)
+                if is_valid_file(fname):
-                if is_valid_file(path):
+                    path = os.path.join(root, fname)
                    item = path, class_index
                    instances.append(item)