Home

Awesome

Co-VeGAN

This is the official implementation code for Co-VeGAN: Complex-Valued Generative Adversarial Network for Compressive Sensing MR Image Reconstruction by Bhavya Vasudeva*, Puneesh Deora*, Saumik Bhattacharya, Pyari Mohan Pradhan (*equal contribution).

Pre-requisites

The code was written with Python 3.6.8 with the following dependencies:

This code has been tested in Ubuntu 16.04.6 LTS with 4 NVIDIA GeForce GTX 1080 Ti GPUs (each with 11 GB RAM).

How to Use

Preparing data

  1. Downloading the dataset:

      MICCAI 2013 dataset:

      MRNet dataset:

      fastMRI dataset:

  1. Run the following command to create the GT dataset:
python dataset_load.py
  1. Run the following command to create the undersampled dataset:
python usamp_data.py
  1. These files would create the training data using MICCAI 2013 dataset. The variables dataset and mode can be changed in both the files to use MRNet or fastMRI datasets, or to create testing data.
  2. The masks folder contains the undersampling masks used in this work. The path for the mask can be modified in usamp_data.py, as required.

Training

  1. Move the files in complexnn folder to the repository folder.
  2. Run the following command to train the model, after checking the names of paths:

      For real-vaued datasets (MICCAI 2013 and MRNet):

python train_model.py

      For complex-vaued dataset (fastMRI):

python train_model_complex.py

Testing

Testing the trained model:

  1. Run the following command to test the model, after checking the names of paths:

      For real-vaued datasets (MICCAI 2013 and MRNet):

python test_model.py

      For complex-vaued dataset (fastMRI):

python test_model_complex.py

Testing the pre-trained model:

  1. The pre-trained generator weights for various undersampling patterns are available at:

      MICCAI 2013:

<p align="center"> <a href="https://drive.google.com/open?id=1WQ92TiBHJXplwwVDZ9jpY-lSBtvV9G6d">30% 1D-G</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/open?id=1u5YC1zJDIk__RDCKrRppHfRXQSiKeupY">30% Radial</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/open?id=1zAxyxs9bpag4iCV2jk4P71RrhO8ry8BS">30% Spiral</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/open?id=1wXC322wti8eucKz9J39wZ2nRrjDezb_f">20% 1D-G</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/open?id=1G60xAEr8na4AbPRtcRAtg6J--Re0j8-s">10% 1D-G</a> </p> &nbsp; &nbsp; &nbsp; fastMRI: <p align="center"> <a href="https://drive.google.com/file/d/1mhR0lTNx3aBmW_rzX9epllhp1VD4CIZN/view?usp=sharing">30% 1D-G</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/file/d/1ysj40CgM5cNLag3kuR2nEqru7691bBpm/view?usp=sharing">30% Radial</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/file/d/1YcdcgfV9KexnDQpdSQJe1u4Ta72SI2z_/view?usp=sharing">30% Spiral</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/file/d/18N5gOGScvEWNCJ4qe32MZwikJrKzGScg/view?usp=sharing">20% 1D-G</a>&nbsp;&nbsp;&bull;&nbsp;&nbsp; <a href="https://drive.google.com/file/d/1o-UncEJG8IGX_90gJDboiPpZiDfvymVs/view?usp=sharing">10% 1D-G</a> </p>
  1. Download the required weights in the repository folder.
  2. Run the following command, after changing the names of paths:

      For MICCAI 2013 dataset:

python test_model.py

      For fastMRI dataset:

python test_model_complex.py

Citation

If you find our research useful, please cite our work.

@InProceedings{Vasudeva_2022_WACV,
    author    = {Vasudeva, Bhavya and Deora, Puneesh and Bhattacharya, Saumik and Pradhan, Pyari Mohan},
    title     = {Compressed Sensing MRI Reconstruction With Co-VeGAN: Complex-Valued Generative Adversarial Network},
    booktitle = {Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision (WACV)},
    month     = {January},
    year      = {2022},
    pages     = {672-681}
}

License

   Copyright 2020 Authors

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.