DIRNet-PyTorch

Reimplementation of DIRNet with PyTorch

(diagram from the reference below)

Usage

# Training
python train.py

Reference

• End-to-End Unsupervised Deformable Image Registration with a Convolutional Neural Network