nnUNet V2修改网络——暴力替换网络为UCTransNet
更换前,要用nnUNet V2跑通所用数据集,证明nnUNet V2、数据集、运行环境等没有问题
阅读nnU-Net V2 的 U-Net结构,初步了解要修改的网络,知己知彼,修改起来才能游刃有余。
UCTransNet是一种创新的医学图像分割网络,它重新思考了U-Net中的跳跃连接设计。该网络以U-Net为基础架构,引入了通道变换器(CTrans)模块,专门用于替代传统的跳跃连接。其核心在于多尺度通道交叉融合变换器(CCT)和通道交叉注意力(CCA)两个子模块。CCT从通道维度出发,利用Transformer的长距离依赖建模优势,对多尺度编码器特征进行融合,为解码器提供更丰富的语义信息。而CCA则进一步优化特征融合过程,使多尺度特征与解码器特征的语义保持一致,从而降低语义差距。这种设计不仅提升了特征融合的精准度,还增强了网络对复杂医学图像中多尺度信息的捕捉能力,为医学图像分割任务提供了更高效、更准确的解决方案,尤其适用于需要精细分割的医学影像分析场景。相比传统U-Net架构,UCTransNet在GlaS、MoNuSeg和Synapse等公共数据集上展示显著