Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "dcb02fcf805524b4df52e31d26953d852bbeb291"
Commit c0a6cfe1 authored by NC Cullen's avatar NC Cullen Committed by Soumith Chintala
Browse files

DS_Store fix in find_classes (#79)

parent 39a50de1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/folder.py
View file @ c0a6cfe1
...@@ -15,7 +15,7 @@ def is_image_file(filename): ...@@ -15,7 +15,7 @@ def is_image_file(filename):
def find_classes(dir): 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() classes.sort()
class_to_idx = {classes[i]: i for i in range(len(classes))} class_to_idx = {classes[i]: i for i in range(len(classes))}
return classes, class_to_idx return classes, class_to_idx
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment