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

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