MGMDcGAN

July 20, 2020 ยท View on GitHub

This is the code of the following paper (tensorflow):

@article{huang2020mgmdcgan,
  title={MGMDcGAN: Medical Image Fusion Using Multi-Generator Multi-Discriminator Conditional Generative Adversarial Network},
  author={Huang, Jun and Le, Zhuliang and Ma, Yong and Fan, Fan and Zhang, Hao and Yang, Lei},
  journal={IEEE Access},
  volume={8},
  pages={55145--55157},
  year={2020},
  publisher={IEEE}
}

It is a unified model for multiple image fusion tasks, including:
MRI-PET medical image fusion
MRI-SPECT medical image fusion
CT-SPECT medical image fusion

Framework:

Training procedure:


Fused results:

MRI-PET:

MRI-SPECT:
CT-SPECT:

To train:

CUDA_VISIBLE_DEVICES=0 python main_medical.py(IS_TRAINING=True)

To test:

CUDA_VISIBLE_DEVICES=0 python main_medical.py(IS_TRAINING=False)

If you have any question, please email to me (lezhuliang@whu.edu.cn).