HowieMa / lstm_multi_modal_UNet

[ICIVC 2019] "LSTM multi-modal UNet for Brain Tumor Segmentation"

lstm_multi_modal_UNet

Implementation of UNet, multi-modal U-Net, LSTM multi-modal Unet for BRATS15 dataset with Pytorch}

Prerequisites

• Python >= 3.6
• Pytorch >= 1.0.0
• SimpleITK
• numpy
• scipy

LSTM multi-modal U-Net

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!

Train

Before training, please remember to change the data_dir

For each folder, to train the model, simply run python main.py

Citation

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}               
}                      

Reference

IVD-Net