/mate

the Pytorch implement of 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.

Primary LanguagePython

Multi-task Autoencoders for Domain Generalizations

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)

Prerequisites

  • Python 3.6+
  • Numpy
  • Pytorch 1.1.0+

Run

python multitaskAutoencoders.py