This is the official code implementation of our WACV paper CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation
This is a followup work of our previous work CAT-Net: A Cross-Slice Attention Transformer Model for Prostate Zonal Segmentation in MRI
If you use this code or the paper in your work, please cite our paper:
@inproceedings{hung2024csam,
title={CSAM: A 2.5 D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation},
author={Hung, Alex Ling Yu and Zheng, Haoxin and Zhao, Kai and Du, Xiaoxi and Pang, Kaifeng and Miao, Qi and Raman, Steven S and Terzopoulos, Demetri and Sung, Kyunghyun},
booktitle={Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision},
pages={5923--5932},
year={2024}
}