摘要
本发明公开了一种基于边缘信息的钢铁表面缺陷显著性检测方法,首先构建基于边缘信息的钢铁表面缺陷显著性检测模型,包括编码器部分、边缘信息提取模块和解码器部分;通过编码器部分提取显著性信息,通过边缘信息提取模块获得边缘信息,最后通过解码器部分融合显著性信息与边缘信息,输出模型预测的结果图。本发明方法通过在编码器部分引入通道注意力机制,对于所提取信息当中重要的部分赋予更大权重,从而使得编码器能更有效地提取信息。本发明还利用编码器输出的信息,提取获得边缘信息并用于指导解码器部分结果图的生成,加强了前景与背景边界区域的检测能力,从而有效地对钢铁表面缺陷区域进行显著性检测。