/CycleMorph

Primary LanguagePython

CycleMorph

Paper

  • CycleMorph: Cycle consistent unsupervised deformable image registration (arXiv.org, Boah Kim et al.)
  • Unsupervised Deformable Image Registration Using Cycle-Consistent CNN (MICCAI 2019, Boah Kim et al.)

Implementation

A PyTorch implementation of deep-learning-based registration. We implemented this code based on voxelMorph and original cycleGAN code. (*Thanks for voxelMorph.) (*Thanks for Jun-Yan Zhu and Taesung Park, and Tongzhou Wang.)

  • Requirements
    • OS : Ubuntu
    • Python 3.6
    • PyTorch 1.4.0

Data

To download the atlas brain and a sample data, visit the Data. The data should be in folder ./data.

Training

  • train.py which is handled by scripts/Brain_train.sh
  • You can run the code by running ./scripts/Brain_train.sh
  • A code for CycleMorph is in models/cycleMorph_model.py.

Testing

  • test.py which is handled by scripts/Brain_test.sh
  • You can run the code by running ./scripts/Brain_test.sh