Implementation of UNet, multi-modal U-Net, LSTM multi-modal Unet for BRATS15 dataset with Pytorch}
- Python >= 3.6
- Pytorch >= 1.0.0
- SimpleITK
- numpy
- scipy
More detail can be found at our origin paper LSTM Multi-modal UNet for brain tumor segmentation
You will also find my presentation slides helpful!
Before training, please remember to change the
data_dir
For each folder, to train the model, simply run
python main.py
If you find this project useful for your research, please use the following BibTeX entry.
@inproceedings{xu2019lstm,
title={LSTM Multi-modal UNet for Brain Tumor Segmentation},
author={Xu, Fan and Ma, Haoyu and Sun, Junxiao and Wu, Rui and Liu, Xu and Kong, Youyong},
booktitle={2019 IEEE 4th International Conference on Image, Vision and Computing (ICIVC)},
pages={236--240},
year={2019},
organization={IEEE}
}