This repo contains code for learning transformations as a generative model by using image registration
-
The models folder contains different model archs based on the dataset being used
- Change the model in train.py for different models
-
There are also different loading functions for different datasets
-
Change folder paths accordingly in constants
-
Use environment.yml to export conda environment (1st Step)
-
libcpab code taken from https://github.com/SkafteNicki/libcpab (2nd Step - follow two steps listed below)
- export PYTHONPATH=$PYTHONPATH:$YOUR_FOLDER_PATH/libcpab
- python setup.py install