71

GitHub - voxelmorph/voxelmorph

 5 years ago
source link: https://github.com/voxelmorph/voxelmorph
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.

README.md

voxelmorph

Notes

  • Code was initially written in python 2.7, and now transfered to 3.5.

  • We are currently cleaning up our code for general use. There are several hard-coded elements related to data preprocessing and format. You will likely need to rewrite some of the data loading code in 'datagenerator.py' for your own datasets.

  • We provide the atlas used in our papers at data/atlas_norm.npz.

Papers

Unsupervised Learning for Fast Probabilistic Diffeomorphic Registration
Adrian V. Dalca, Guha Balakrishnan, John Guttag, Mert R. Sabuncu
MICCAI 2018. eprint arXiv:1805.04605

An Unsupervised Learning Model for Deformable Medical Image Registration
Guha Balakrishnan, Amy Zhao, Mert R. Sabuncu, John Guttag, Adrian V. Dalca
CVPR 2018. eprint arXiv:1802.02604

Instructions

Training:

  1. Change base_data_dir in train.py to the location of your image files.
  2. Run train.py [model_name] [gpu-id]

Testing (Dice scores):

Put test filenames in data/test_examples.txt, and anatomical labels in data/test_labels.mat.

  1. Run test.py [model_name] [gpu-id] [iter-num]

Contact:

[email protected]


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK