(dir)
| 28 | |
| 29 | |
| 30 | def find_classes(dir): |
| 31 | classes = [d for d in os.listdir(dir) if os.path.isdir(os.path.join(dir, d))] |
| 32 | classes.sort() |
| 33 | class_to_idx = {classes[i]: i for i in range(len(classes))} |
| 34 | return classes, class_to_idx |
| 35 | |
| 36 | |
| 37 | def make_dataset(dir, class_to_idx, extensions): |