(root='./data', train=True, transform=None)
| 134 | |
| 135 | |
| 136 | def datasetImageNetMini(root='./data', train=True, transform=None): |
| 137 | dataset = datasetImageNet(root=root, train=train, transform=transform) |
| 138 | ''' imagenet-mini is a subset of the first 100 classes of ImageNet ''' |
| 139 | idx = np.where( np.array(dataset.targets) < 100 )[0] |
| 140 | dataset.samples = [ dataset.samples[ii] for ii in idx ] |
| 141 | dataset.targets = [ dataset.targets[ii] for ii in idx ] |
| 142 | return dataset |
| 143 | |
| 144 | |
| 145 | def get_transforms(dataset, train=True, is_tensor=True): |
no test coverage detected