摘要:针对现有方法分割弱边缘铸件 CT 图像难度大、精度低、鲁棒性差的问题,提出一种融合残差模块与混合注意力机制的 U 型网络分割算法(AttRes-U-Nets)。 该算法以 U-Net 网络为基础,首先构建深度残差网络 ResNets 作为算法的编码网络,解决 传统 U-Net 网络特征提取能力不足的问题;然后,引入改进后的混合注意力机制,突出分割目标区域与通道的特征响应,提高网 络灵敏度;最后,将 Focal loss 与 Dice loss 结合为一种新损失函数 FD loss 缓解样本不平衡带来的负面影响。 使用 120 阀体数据 集对算法性能进行验证,实验结果表明,本文算法对铸件分割的像素准确率(PA)和交互比( IoU)分别达到 98. 72% 和 97. 40% , 优于传统 U-Net 算法与其他主流语义分割算法,为弱边缘分割提供了新思路。