vmmm123 / mate Goto Github PK

This code contains the Pytorch implementation of the Multitask Autoencoder (MTAE) algorithm based on the following paper:

M. Ghifary, W. B. Kleijn, M. Zhang, D. Balduzzi. Domain Generalization for Object Recognition with Multi-task Autoencoders, accepted in International Conference on Computer Vision (ICCV 2015), Santiago, Chile.

Please cite the above paper when using this code.

Notes:

• Currently only works for the provided MNIST dataset (with 6 rotated views)

• Python 3.6+
• Numpy
• Pytorch 1.1.0+

python multitaskAutoencoders.py