This dataset contains over 1,000 images of horses and zebras respectively. This is intended to be used with image translation models, such as CycleGANs, although you can also use it with regular GANs as well. This dataset was ported from a CycleGAN implementation by the authors Zhu et al. You can obtain the original code here.
The default image size is 256x256, but you can also specify different size during training, e.g.
mantra train cyclegan --dataset horse2zebra --image-dim 128 128
To import this dataset to your project, run:
mantra import RJT1990/data/horse2zebra
Example usage in code:
from data.horse2zebra.data import Horse2Zebra
d = Horse2Zebra()
print(d.Y.__class__) ## <class 'numpy.ndarray'> contains zebras
print(d.X.__class__) ## <class 'numpy.ndarray'> contains horses