江汉大学学报(自然科学版) ›› 2024, Vol. 52 ›› Issue (2): 56-67.doi: 10.16389/j.cnki.cn42-1737/n.2024.02.007
• 计算机图形图像学 • 上一篇
张志玮,叶 曦*,杨志红
ZHANG Zhiwei,YE Xi*,YANG Zhihong
摘要: 目前主流图像分割算法在分割边界上对特征相似而类别不同的像素鉴别能力不佳,从而 影响了分割精度。设计了一种基于曼哈顿距离自注意力机制的 U-Net3+图像分割算法,通过 关注不同特征点之间信息表征的差异程度来对大范围上下文信息关系进行建模,增强算法对 特征相似而类别不同的像素的鉴别能力和对全局关系的学习能力;再通过 U-Net3+的全尺度 跳跃连接结构将不同尺度的特征相融合,为算法提供更多尺度的上下文信息,使分割算法兼顾细 节信息和全局关系。使用 COVID-19 CT 数据集对该算法进行实验测试,结果表明,引入基于 曼哈顿距离自注意力机制后 U-Net3+的 Dice 和 IoU 指标分别提升了 2. 79% 和 3. 17%,对比使 用多头自注意力机制的 U-Net3+分别提升了 1. 06% 和 1. 02%,证明了该算法的有效性和优越性。
中图分类号: