(root='./data', train=True, transform=None)
| 128 | |
| 129 | |
| 130 | def datasetImageNet(root='./data', train=True, transform=None): |
| 131 | if train: root = os.path.join(root, 'ILSVRC2012_img_train') |
| 132 | else: root = os.path.join(root, 'ILSVRC2012_img_val') |
| 133 | return torchvision.datasets.ImageFolder(root=root, transform=transform) |
| 134 | |
| 135 | |
| 136 | def datasetImageNetMini(root='./data', train=True, transform=None): |
no outgoing calls
no test coverage detected