The code is derivated from the following paper and GitHub repository:
Paper: Wang, Fei, et al. "Residual attention network for image classification." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017. https://ieeexplore.ieee.org/document/8100166
GitHub Repository: https://github.com/tengshaofeng/ResidualAttentionNetwork-pytorch
I extended the 2D network into the 3D network to deal with Overall Survival (OS) classification of patients with Primary Central Neural System Lymphoma (PSNSL) using 3D Magnetic Resonance Imaging (MRI).