Produces [image, label] in Cifar10 dataset, image is 32x32x3 in the range [0,255]. label is an int.
| 168 | |
| 169 | |
| 170 | class Cifar10(CifarBase): |
| 171 | """ |
| 172 | Produces [image, label] in Cifar10 dataset, |
| 173 | image is 32x32x3 in the range [0,255]. |
| 174 | label is an int. |
| 175 | """ |
| 176 | def __init__(self, train_or_test, shuffle=None, dir=None): |
| 177 | """ |
| 178 | Args: |
| 179 | train_or_test (str): either 'train' or 'test'. |
| 180 | shuffle (bool): shuffle the dataset, default to shuffle in training |
| 181 | """ |
| 182 | super(Cifar10, self).__init__(train_or_test, shuffle, dir, 10) |
| 183 | |
| 184 | |
| 185 | class Cifar100(CifarBase): |
no outgoing calls
no test coverage detected
searching dependent graphs…