This repo is the official implementation for Category Guided Attention Network for Brain Tumor Segmentation in MRI. The multimodal brain tumor dataset (BraTS 2019) could be acquired from here.
The Unet_Both model can be found here.
- python 3.7
- pytorch 1.6.0
- torchvision 0.7.0
- pickle
- nibabel