Skip to content

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

Notifications You must be signed in to change notification settings

HowieMa/lstm_multi_modal_UNet

Repository files navigation

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

About

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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages