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 LSTM Multi-modal UNet for brain tumor segmentation You will also find my presentation slides helpful! If you use this code in your research, please cite it, thank you!

Train

Before training, please remember to change the data_dir

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

Reference

IVD-Net