有效

基于改进YOLOv5的表面缺陷检测方法

杨文、汤臣薇、吉昱阳、郭美彤、刘沛东、程锦国
四川大学

摘要

本发明公开了一种基于改进YOLOv5的表面缺陷检测方法,涉及金属检测领域。本发明基于YOLOv5s神经网络模型结构进行改进,在原有Backbone主干网络部分尾部增添使用了C3STR模块,通过C3STR模块对缺陷特征进行捕获强化,从而提取出更加丰富的图像缺陷特征,提高对图像中缺陷识别的准确性,并且能够降低对缺陷特征进行提取时的计算量,使得具有轻量性的特点。在将缺陷特征提取处标注框信息时,使用Coordinate Attention坐标注意力模块先对通道、方向和位置等感知信息进行捕获,能够帮助YOLOv5s的Head检测头部分对缺陷特征进行定位和识别,具有灵活、轻量和快速等优点。