融合注意力机制的金属缺陷图像分割方法
DOI:
CSTR:
作者:
作者单位:

(1.昆明理工大学 信息工程与自动化学院,云南 昆明 650500; 2.武钢集团昆明钢铁股份有限公司安宁公司,云南 昆明 650302)

作者简介:

杨晓洪(1964-),女,硕士研究生,高级工程师,主要从事 控制工程与理论控制,大数据应用,图像处理等方面的研究.

通讯作者:

中图分类号:

基金项目:

国家重点研发计划(2017YFB0306405)、云南省重点研发计划(2018BA070)和国家自然科学基 金(61364008)资助项目 (1.昆明理工大学 信息工程与自动化学院,云南 昆明 650500; 2.武钢集团昆明钢铁股份有限公司安宁公司,云南 昆明 650302)


Metal defect image segmentation algorithm combined with attention mechanism
Author:
Affiliation:

(1.Faculty of Information Engineering and Automation,Kunming University of Sci ence and Technology,Yunnan Kunming,650500,China; 2.WISCO Group Kunming Iron&Stee l Co,Ltd,Anning Company,Yunnan Kunming,650302,China)

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

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

    Abstract:

    Due to the characteristics of metal surface defect images,effective an d accurate segmentation is a major challenge in image processing tasks.In order to obtain defect type,size and location information,this paper proposes a met al defect image segmentation network that incorporates attention mechanism.The network is divided into two paths.The semantic information path is mainly compo sed of a convolutional network composed of residual blocks to obtain a feature m ap.During the sampling process,the attention mechanism is integrated step by s tep to enhance the contrast between the feature and the background.The side roa d path design attention mechanism module obtains the weight map of the location information,and then combines the feature map of the same size with the weight map,and combines multi-scale features through the spatial pyramid.Experimenta l results show that the algorithm can improve the segmentation accuracy of metal surface defect images.

    参考文献
    相似文献
    引证文献
引用本文

赵鹤,杨晓洪,杨奇,尹丽琼.融合注意力机制的金属缺陷图像分割方法[J].光电子激光,2021,32(4):403~408

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2020-12-02
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2021-04-22
  • 出版日期:
文章二维码