CAT-Net: A Cross-Slice Attention Transformer Model for Prostate Zonal Segmentation in MRI

The official code for our TMI paper CAT-Net: A Cross-Slice Attention Transformer Model for Prostate Zonal Segmentation in MRI

There is a followup work of this paper, which is accepted at WACV, CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation

Credit

If you use this code or the paper in your work, please cite our paper:

@article{hung2022cat,
  title={CAT-Net: A cross-slice attention transformer model for prostate zonal segmentation in MRI},
  author={Hung, Alex Ling Yu and Zheng, Haoxin and Miao, Qi and Raman, Steven S and Terzopoulos, Demetri and Sung, Kyunghyun},
  journal={IEEE transactions on medical imaging},
  volume={42},
  number={1},
  pages={291--303},
  year={2022},
  publisher={IEEE}
}