ICW-GANs
This is a Tensorflow implementation of my paper:
FMRI data augmentation via synthesis, The IEEE International Symposium on Biomedical Imaging (ISBI'19)
Peiye Zhuang, Alexander Schwing, and Sanmi Koyejo
Prerequisites
- Tensorflow 1.x
- Python3
- NVIDIA GPU + CUDA CuDNN
- Scipy 1.1.0, Nilearn etc.
Data
We used a public dataset BrainPedia 1952. You may either download the dataset by yourself or use our preprocessed data on GoogleDrive.
Pretrained models
You may download our pretrained model checkpoint on GoogleDrive.
Training & testing
python icw_gans.py
You may change default parameter settings in the argparse. We did not write an independent python file for testing. Instead, we used the function save_test_img
in the code to save test images after amount of training epochs.
Citation
If you use this code for your research, please cite our paper:
@article{icwgans,
title={FMRI data augmentation via synthesis},
author={Peiye Zhuang, Alexander Schwing, Sanmi Koyejo},
journal={The IEEE International Symposium on Biomedical Imaging (ISBI)},
year={2019}
}