Shape Registration in the Time of Transformers [Paper]

Shape Registration in the Time of Transformers
Giovanni Trappolini, Luca Cosmo, Luca Moschella, Riccardo Marin, Simone Melzi, Emanuele Rodolà
In Neurips, 2021

Code Usage & Reproducible Results

Download the dataset folder here and put it under transmatching/test/ .

Install environment and dependencies:

conda create -n transformer_matching
conda activate transformer_matching

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install -c conda-forge igl
pip install ./transmatching


cd transmatching/test

To train the model just run

python train.py

To test the model please look into evaluation