Home

Awesome

DeepJDOT

This repository contains the codes of the following paper

BB Damodaran, B Kellenberger, R Flamary, D Tuia, N Courty, "DeepJDOT:Deep Joint Distribution Optimal Transport for Unsupervised Domain Adaptation", in European Conference on Computer Vision 2018 (ECCV-2018).

Dependencies

In order to run, the code requires the following Python modules:

Modules

To run the DeepJDOT on the sample 2D dataset, please see or run the "deepjdot_demo.py".

To run on the real data set: SVHN --> MNIST, please see "deepjdot_svhn_mnist.py". The default task is to do classification but you can turn on regression demo by setting do_reg to True. If you do not want to wait long training time you can set small_model to True.

For regression demo, each image will be randomly rotated around its center, and then the label will be the angle rotated. The angle will be scaled to [0, 1]. The model needs to predict how much the image is rotated.

I suggest you run the demo files inside Spyder or any interactive python IDE so that you can investigate each cell denoted by #%% lines and understand the code better.