CIFAR-10
The CIFAR-10 dataset consists of 60000 32x32 colour images in 10 classes, with 6000 images per class.
To train with an example CNN classification model:
mantra train cnn_example --dataset cifar10 --image-dim 32 32
Importing
To import this dataset to your project, run:
mantra import RJT1990/data/cifar10
Usage
Example usage in code:
from data.cifar10.data import CIFAR10
d = CIFAR10()
print(d.X.__class__) ## <class 'numpy.ndarray'>
print(d.y.__class__) ## <class 'numpy.ndarray'>