DiffuseMorph: Unsupervised Deformable Image Registration Using Diffusion Model

This repository is the official implementation of "DiffuseMorph: Unsupervised Deformable Image Registration Using Diffusion Model".

Requirements

• OS : Ubuntu
• Python >= 3.6
• PyTorch >= 1.4.0

Data

In our experiments, we used the following datasets:

• 2D facial expression images: RaFD dataset
• 3D cardiac MR images: ACDC dataset
• 3D brain MR images: OASIS-3 dataset

Training

To train our model for 2D image registration, run this command:

python3 main_2D.py -p train -c config/diffuseMorph_train_2D.json

To train our model for 3D image registration, run this command:

python3 main_3D.py -p train -c config/diffuseMorph_train_3D.json

Test

To test the trained our model for 2D image registration, run:

python3 main_2D.py -p test -c config/diffuseMorph_test_2D.json

To test the trained our model for 3D image registration, run:

python3 main_3D.py -p test -c config/diffuseMorph_test_3D.json