[发明专利]一种输电线路金具及绝缘子检测方法

权利要求书

1.一种输电线路金具及绝缘子检测方法，其特征在于，包括：

将标注样本图像输入特征提取网络，得到第一特征图，所述第一特征图包含三种不同尺度的特征图；

将第一特征图输入CBAM注意力模块，得到第二特征图，所述第二特征图包含三种不同尺度的特征图；

将第二特征图输入特征融合网络，得到第三特征图，所述第三特征图包含三种不同尺度的特征图；

将第三特征图输入到YOLO head，得到检测结果。

2.根据权利要求1所述方法，其特征在于，所述标注样本图像是使用labelimg标注工具对整理好的航拍图像进行标注得到的。

3.根据权利要求1所述方法，其特征在于，所述特征提取网络采用CSPDarknet53。

4.根据权利要求1所述方法，其特征在于，CBAM注意力模块在特征提取网络和特征融合网络之间，从空间和通道两个层面分别关注目标区域，增强相关特征、抑制无关特征，帮助信息在网络中的传递。

5.根据权利要求1所述方法，其特征在于，CBAM注意力模块使用CBAM注意力机制，CBAM注意力机制整体分为两部分，分别为通道注意力部分和空间注意力部分，CBAM注意力模块会将第一特征图的空间维度进行压缩，聚集空间信息，首先通过全局池化和最大池化收集独特的目标特征，用于推断出更为准确的通道关注，提升网络表达能力，池化操作会得到两个不同的1×1×c的特征图，其中c代表通道数，再将这两个特征图送入一个两层的神经网络，最后经sigmoid激活函数完成通道注意力的生成，得到通道注意力M_c(F₀)：

M_c(F₀)＝δ(MLP(AvgPool(F₀))+MLP(MaxPool(F₀))) (1)

式(1)中F₀表示输入的第一特征图，δ和MLP分别表示sigmoid激活函数和神经网络运算，AvgPool和MaxPool为平均池化和最大池化。

输出的通道注意力与输入的特征图F₀相乘得到第四特征图作为空间注意力的输入特征，空间注意力更加关注特征信息属于哪一类，与通道注意力相辅相成，首先沿着通道方向进行平均池化和最大池化操作，聚合图中的通道信息，得到两个H×W×1的特征图，H与W分别表示特征图的高和宽，再将这两个特征图连接生成一个特征描述，然后经卷积运算再通过sigmoid激活函数生成空间注意力Ms(F₁)如下所式：

M_s(F₁)＝δ(T[AvgPool(F₁),MaxPool(F₁)]) (2)

式(2)中F₁表示输入的第四特征图，T为卷积操作，最终输出的第二特征图用F₂表示：

。

6.根据权利要求1所述方法，其特征在于，所述特征融合网络为PANet特征融合网络，所述PANet特征融合网络是在特征金字塔的基础上增加了一个自底向上的路径增强，并且将最初提取的特征与最终提取的特征相融合，以缩短底层特征与顶层特征之间的信息路径，利用底层的特征信息增强整体特征，提高其信息的利用率。