Demo code for "Joint Learning of Motion Estimation and Segmentation for Cardiac MR Image Sequences"

Lasagne and theano implementation of the framework.

main.py ==> main training file

test_prediction.py ==> applies joint prediction and visualisation

models ==> proposed network and layers

dataio ==> includes loading images, data_augmentation, etc

utils ==> metrics and visualisation

model ==> Model parameters

test ==> One test sample

pytorch_version ==> pytorch implementation of the models

Citation and Acknowledgement

If you use the code for your work, or if you found the code useful, please cite the following works:

Qin, C., Bai, W., Schlemper, J., Petersen, S.E., Piechnik, S.K., Neubauer, S. and Rueckert, D. Joint learning of motion estimation and segmentation for cardiac MR image sequences. In International Conference on Medical Image Computing and Computer-Assisted Intervention (MICCAI), 2018: 472-480.

C. Qin, W. Bai, J. Schlemper, S. Petersen, S. Piechnik, S. Neubauer and D. Rueckert. Joint Motion Estimation and Segmentation from Undersampled Cardiac MR Image. International Workshop on Machine Learning for Medical Image Reconstruction, 2018: 55-63.