基于坐标注意力机制与解耦头结构的YOLOv5带钢表面检测方法

本发明提供基于坐标注意力机制与解耦头结构的YOLOv5带钢表面检测方法，涉及表面缺陷检测技术领域，包括，S1.获取表面缺陷的原始图像，在YOLOv5模型的主干网络部分引入坐标注意力机制，通过沿水平方向和垂直方向分别进行两个一维全局池化操作，从而生成综合的方向感知特征图。该基于坐标注意力机制与解耦头结构的YOLOv5带钢表面检测方法，通过反向传播显著性区域增强机制，能够动态检测图像中的显著性区域并进行加权，从而增强图像中缺陷区域的特征表达。这一机制使得模型在进行多尺度特征融合时，能够自动调整对重要区域的关注度，确保模型在面对多尺度和复杂缺陷时，既能够精准识别小缺陷，又能保证在大范围缺陷检测中的鲁棒性。