基于U2-Net的金属表面缺陷检测算法

Metal surface defect detection algorithm based on U2-Net

金属表面缺陷检测旨在通过合理的算法判断工业生产中金属材料的表面图像中是否存在缺陷,是计算机视觉领域在工业应用中的重要研究内容,但现有的基于分割的金属表面缺陷检测算法存在抗干扰能力弱、容易背景误判和检测粒度不够细致等问题.针对具有各种干扰因子的金属表面图像,提出一种基于U2-Net的金属表面缺陷检测算法Attention-U2Net.首先,为了解决实际生产中金属表面图像噪点过多导致背景误判和采样层获取信息不够细致的问题,设计U型注意力编码模块,可以在编码时增大缺陷区域权重的同时,抑制背景噪声;然后,为了解决图像中存在的难检测样本和边界复杂问题,设计具有权重的损失函数并结合多层次监督方法,使检测算法更关注难检测样本和边界像素点,提高预测准确度;最后,使用根据图像结果自动计算灰度阈值进行结果优化的算法,最终生成高质量缺陷预测图.与六种常用的缺陷检测领域的像素分割方法在金属表面缺陷公开数据集上进行比较实验,实验结果表明,提出的算法的像素准确率、查准率、查全率、F-score、平均绝对误差和均方误差均取得了优良的结果,证明该算法拥有较强的抗干扰因子能力,最终缺陷预测图像素准确率高,边界明显.

Metal surface defect detection aims to judge whether there are defects in the surface image of metal materials in industrial production through a reasonable algorithm,and is an important research content in the field of computer vision applied in industry.However,the existing segmentation-based metal surface defect detection algorithms still have problems such as weak anti-interference ability,easy background misjudgment and insufficient detection granularity.For metal surface images with various interference factors,this paper proposes a U²-Net-based metal surface defect detection algorithm Attention-U'Net.First of all,in order to solve the problem of misjudgment of background and insufficient information obtained by sampling layer due to excessive noise of metal surface image in actual production,a U-shaped attention coding module is designed for suppressing background noise.Then,in order to solve the complex problems of difficult-to-detect samples and boundaries in images,a weighted loss function is designed and combined with a multi-level supervision method,so that the detection algorithm pays more attention to difficult-to-detect samples and boundary pixels and improves the prediction accuracy.Finally,an algorithm that automatically calculates the grayscale threshold according to the image results is used to optimize the results,and a high-quality defect prediction map is generated finally.By comparing with six pixel segmentation methods commonly used in the field of defect detection,the experimental results on the public data set of metal surface defects show that the algorithm has good performance.The experimental results show that the algorithm has strong anti-interference factor ability,and the final defect prediction image has high pixel accuracy and obvious boundaries.