(self, root, loader, extensions, ann_file='', img_prefix='', transform=None, target_transform=None,
cache_mode="no")
| 91 | """ |
| 92 | |
| 93 | def __init__(self, root, loader, extensions, ann_file='', img_prefix='', transform=None, target_transform=None, |
| 94 | cache_mode="no"): |
| 95 | # image folder mode |
| 96 | if ann_file == '': |
| 97 | _, class_to_idx = find_classes(root) |
| 98 | samples = make_dataset(root, class_to_idx, extensions) |
| 99 | # zip mode |
| 100 | else: |
| 101 | samples = make_dataset_with_ann(os.path.join(root, ann_file), |
| 102 | os.path.join(root, img_prefix), |
| 103 | extensions) |
| 104 | |
| 105 | if len(samples) == 0: |
| 106 | raise (RuntimeError("Found 0 files in subfolders of: " + root + "\n" + |
| 107 | "Supported extensions are: " + ",".join(extensions))) |
| 108 | |
| 109 | self.root = root |
| 110 | self.loader = loader |
| 111 | self.extensions = extensions |
| 112 | |
| 113 | self.samples = samples |
| 114 | self.labels = [y_1k for _, y_1k in samples] |
| 115 | self.classes = list(set(self.labels)) |
| 116 | |
| 117 | self.transform = transform |
| 118 | self.target_transform = target_transform |
| 119 | |
| 120 | self.cache_mode = cache_mode |
| 121 | if self.cache_mode != "no": |
| 122 | self.init_cache() |
| 123 | |
| 124 | def init_cache(self): |
| 125 | assert self.cache_mode in ["part", "full"] |
no test coverage detected