融合注意力机制的金属缺陷图像分割方法

摘要：

由于金属表面缺陷图像的特性,有效精确分割是图像处理任务中的一大挑战.为了获得缺陷的类型、大小及位置信息,本文提出一种融合注意力机制的金属缺陷图像分割网络.该网络分为两条路径,语义信息路径主要由残差块构成的卷积网络获得特征图,采样过程中分步融合注意力机制以增强特征与背景对比度.旁路路径设计注意力机制模块获得位置信息的权重图,后将同尺寸的特征图与权重图融合,通过空间金字塔结合多尺度特征.实验结果表明,运用该算法可以提高金属表面缺陷图像的分割精度.

关键词：语义分割、膨胀卷积、注意力机制、特征融合

所属期刊栏目：32

分类号：TP391(计算技术、计算机技术)

资助基金：国家重点研发计划;云南省重点研发计划;国家自然科学基金

在线出版日期：2021-05-28（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：403-408