Skip to content

timeanddoctor/Fully_Unsupervised_CNN_Registration_Keras

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

66 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

2D/3D Medical Imaging Registration via Fully Unsupervised ConvNet

This is a Keras/Tensorflow implementation of my paper:

Chen, Junyu, et al. "Generating Anthropomorphic Phantoms Using Fully Unsupervised Deformable Image Registration with Convolutional Neural Networks." Medical Physics. Accepted Author Manuscript. doi:10.1002/mp.14545. 2020.

Chen, Junyu, et al. "A fully unsupervised approach to create patient-like phantoms via Convolutional neural networks." Journal of Nuclear Medicine 61.supplement 1 (2020): 522-522.

For most updated scripts, see "CNN_MedPhy/" folder.

We treat CNN as an optimization tool that iteratively minimizes the loss function via reparametrization in each iteration. This means that the algorithm is fully unsupervised and thus no prior training is required. The registration loss function is defined as:

, where I_d and I_f are, respectively, the deformed and the fixed image, L_sim represents the loss function for image similarity, and R represents the regularization applied on the deformation field.

The effects of different loss functions:

The effects of different regularizations:

Sample results for XCAT phantom to and patient CT registration:

Some deformed phantom and SPECT simulations:

If you find this code is useful in your research, please consider to cite:

@article{chen2020phantoms,
author = {Chen, Junyu and Li, Ye and Du, Yong and Frey, Eric C.},
title = {Generating Anthropomorphic Phantoms Using Fully Unsupervised Deformable Image Registration with Convolutional Neural Networks},
journal = {Medical Physics},
volume = {n/a},
number = {n/a},
pages = {},
doi = {10.1002/mp.14545},
url = {https://aapm.onlinelibrary.wiley.com/doi/abs/10.1002/mp.14545},
eprint = {https://aapm.onlinelibrary.wiley.com/doi/pdf/10.1002/mp.14545},
}

About

Fully unsupervised 2D/3D image registration with ConvNet.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%