[发明专利]一种输电线路防震锤锈蚀缺陷检测方法和系统

权利要求书

1.一种输电线路防震锤锈蚀缺陷检测方法，其特征在于，包括：

构建输电线路防震锤图像样本数据集，其中，输电线路防震锤图像样本数据集包括锈蚀缺陷图像数据集和正常图像数据集；

利用ESPNetv2模型中的EESP模块替换YOLOv8s检测模型中C2f模块的CBS模块，利用PReLU激活函数替换YOLOv8s检测模型中主干网络中起到下采样作用的卷积模块中的SiLU激活函数，利用下采样操作和1×1的标准卷积代替YOLOv8s检测模型特征金字塔中的头部网络的CBS模块，得到改进的YOLOv8s检测模型，EESP模块中的并行深度空洞可分离卷模块个数为4个，空洞率分别为3、5、7和3；

根据输电线路防震锤图像样本数据集对改进的YOLOv8s检测模型进行训练，得到目标检测模型；

将待识别的含有防震锤的输电线路图像输入到目标检测模型中，得到输电线路防震锤锈蚀缺陷检测结果。

2.根据权利要求1所述的输电线路防震锤锈蚀缺陷检测方法，其特征在于，构建输电线路防震锤图像样本数据集，包括：

获取输电线路巡检图像中含有防震锤的图像，并对图像进行预处理，构建真实输电线路防震锤图像数据集；

基于真实输电线路防震锤图像数据集，对生成对抗网络模型进行训练，基于真实输电线路防震锤图像数据集中含有防震锤锈蚀缺陷的图像，通过已训练好的生成对抗网络中的生成网络生成含有防震锤锈蚀缺陷的输电线路防震锤图像；

将真实输电线路防震锤图像数据集中含有防震锤锈蚀缺陷的图像和由生成网络生成的含有防震锤锈蚀缺陷的输电线路防震锤图像加入锈蚀缺陷图像数据集，将真实输电线路防震锤图像数据集中不含有防震锤锈蚀缺陷的图像加入正常图像数据集。

3.根据权利要求2所述的输电线路防震锤锈蚀缺陷检测方法，其特征在于，根据输电线路防震锤图像样本数据集对改进的YOLOv8s检测模型进行训练，得到目标检测模型，包括：

利用Gogle Open Image公共数据集对改进的YOLOv8s检测模型进行预训练，将预训练得到的模型参数作为改进的YOLOv8s检测模型的初始参数；

利用输电线路防震锤图像样本数据集对预训练后的改进的YOLOv8s检测模型进行训练，得到目标检测模型。

4.根据权利要求2所述的输电线路防震锤锈蚀缺陷检测方法，其特征在于，生成对抗网络模型为SinGAN网络模型。

5.一种输电线路防震锤锈蚀缺陷检测系统，其特征在于，包括：

数据集构建模块，用于构建输电线路防震锤图像样本数据集，其中，输电线路防震锤图像样本数据集包括锈蚀缺陷图像数据集和正常图像数据集；

改进检测模型构建模块，用于利用ESPNetv2模型中的EESP模块替换YOLOv8s检测模型中C2f模块的CBS模块，利用PReLU激活函数替换YOLOv8s检测模型中主干网络中起到下采样作用的卷积模块中的SiLU激活函数，利用下采样操作和1×1的标准卷积代替YOLOv8s检测模型特征金字塔中的头部网络的CBS模块，得到改进的YOLOv8s检测模型，ESPNetv2模型中的EESP模块中的并行深度空洞可分离卷模块个数为4个，空洞率分别为3、5、7和3；

检测模型训练模块，用于根据输电线路防震锤图像样本数据集对改进的YOLOv8s检测模型进行训练，得到目标检测模型；

锈蚀缺陷识别模块，用于将待识别的含有防震锤的输电线路图像输入到目标检测模型中，得到输电线路防震锤锈蚀缺陷检测结果。