医用变压器 论文Pytorch代码 关于此仓库: 此存储库托管以下网络的代码: 门控轴向注意U-Net 医学博士 介绍 提议用于视觉应用的大多数现有的基于变压器的网络体系结构都需要大规模的数据集才能正确地进行训练。 但是,与用于视觉应用的数据集相比,对于医学成像而言,数据样本的数量相对较少,因此难以有效地训练用于医学应用的变压器。 为此,我们提出了门控轴向注意力模型,该模型通过在自我注意力模块中引入其他控制机制来扩展现有体系结构。 此外,为了有效地在医学图像上训练模型,我们提出了局部全局训练策略(LoGo),可以进一步提高性能。 具体来说,我们在整个图像和补丁上进行操作,以分别学习全局和局部特征。 拟议的医疗变压器(MedT)在门控轴向注意U-Net上使用LoGo培训策略。 使用代码: 克隆此存储库: git clone https://github.com/jeya-maria