改进型YOLOv5s的瓷砖表面缺陷检测方法

作者：吴航星; 张宏怡^*; 谭湘琼; 林海峰

来源：厦门理工学院学报, 2023, 31(05): 66-74.

DOI：10.19697/j.cnki.1673-4432.202305010

• 瓷砖表面
• 缺陷检测
• 图像处理
• YOLOv5
• 特征提取
• 多尺度检测

针对瓷砖质检过程中存在瓷砖缺陷漏检及检测精度低等问题，提出一种改进型YOLOv5s的瓷砖表面缺陷检测方法。该方法在自适应锚框匹配算法中引入CIoU (complete intersection over union)度量，解决原始K-Means聚类算法中欧式距离无法聚类出宽高比合适的锚框问题；在特征提取结构中，引入Res2Net残差单元，在通道上进行分组级联特征提取，实现模块级别的多尺度检测；在缺陷检测阶段使用滑窗方法进行高分辨率瓷砖图像的缺陷检测。实验结果表明，使用该方法后瓷砖表面缺陷检测的精度由73.8%提升至80.0%；召回率由69.0%提升至78.1%;50%交并比精度均值（mean average precision at 50%IoU,mAP50）由72.6%提升至80.5%；与其他主流算法相比，该算法具有更好的检测效果和鲁棒性。

• 出版日期2023
• 单位厦门理工学院