Skip to content

bbdamodaran/deepJDOT

Repository files navigation

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:

  • Numpy
  • Matplotlib
  • POT (Python Optimal Transport library)
  • keras with tensorflow backend (preferably GPU version)
  • imutils (only for rotating images in regression demo)
  • scikit-learn (for scoring functions)

Modules

  • Deepjdot - module contains the implementation of the DeepJDOT
  • dnn - import necessary functions from keras
  • deepjdot_demo - DeepJDOT on the sample dataset
  • deepjdot_svhn_mnist - DeepJDOT on SVHN & MNIST dataset

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.

About

Implementation of DeepJDOT in Keras

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages