Official repository for the paper Data augmentation with Mobius transformations by Sharon Zhou, Jiequan Zhang, Hang Jiang, Torbjörn Lundh, and Andrew Ng.

Apply Mobius transformations to your images!

Original Images:

Outputs:

Change the filename in Mobius_demo.ipynb and you will be able to apply Mobius transformations to your own images! :)

Navigate to mobius_data_augmentation. Make sure you have downloaded the respective datasets to their folders specified in train.py. Feel free to experiment with different parameters.

Experimental settings in the paper (to be run in the mobius_data_augmentation folder):

python train.py --data_augmentation mix3 --name mobius

python train.py --data_augmentation mix3 --cutout --name mobius-and-cutout

python train.py --data_augmentation regular --cutout --name cutout

python train.py --data_augmentation regular --name crop-and-flip