Vibashan / 3D_Attention_UNet

Official Pytorch Code base for 3D Attention Unet [MICCAI 2020]

3D_Attention_UNet

This repository contains the official implementation of the paper "Brain Tumor Segmentation and Survival Prediction using 3D Attention UNet" preprint and in workshop Proceedings

The baseline implementation of UNet3D is adopted from
SSD: https://github.com/wolny/pytorch-3dunet

Dataset

Multimodal Brain Tumor Segmentation Challenge 2019 (BraTS)

Citation

If you use this code for your research, please cite our paper.

@inproceedings{islam2019brain,
  title={Brain tumor segmentation and survival prediction using 3D attention UNet},
  author={Islam, Mobarakol and Vibashan, VS and Jose, V Jeya Maria and Wijethilake, Navodini and Utkarsh, Uppal and Ren, Hongliang},
  booktitle={International MICCAI Brainlesion Workshop},
  pages={262--272},
  year={2019},
  organization={Springer}
}