This dataset contains over 200,000 images of celebrity faces.
The default image size is 128x128, but you can also specify different size during training, e.g.
mantra train sagan --dataset celeba --image-dim 256 256
To import this dataset to your project, run:
mantra import RJT1990/data/celeba
This is a large dataset, so the first time you train with it, it may take a while to version and upload to the cloud. But after you use it once, it should be quick to use.
Example usage in code:
from data.celeba.data import CelebA
d = CelebA()
print(d.X.__class__) ## <class 'numpy.ndarray'>
print(d.X.shape) ## (5680, 128, 128, 3)