PDF(4682 KB)
基于改进YOLOv5s的旋转绝缘子缺陷检测
朱文凯, 李运堂, 张坤, 李恒杰, 冯娟, 金杰, 章聪, 陈源, 李孝禄, 白金鹏
计量学报 ›› 2026, Vol. 47 ›› Issue (2) : 271-282.
PDF(4682 KB)
PDF(4682 KB)
基于改进YOLOv5s的旋转绝缘子缺陷检测
Defect Detection of Rotating Insulators Based on Improved YOLOv5s
针对航拍角度导致的水平矩形框中的绝缘子检测背景区域过大的问题,采用旋转目标框替代水平矩形框对绝缘子进行检测,构建Ghost-YOLOv5s网络,实现旋转绝缘子缺陷高精度、快速检测。将轻量化的GhostNet作为YOLOv5s主干网络,减少网络参数,提高检测速度;颈部网络采用双向特征金字塔结构,并引入可学习的通道权重进行自适应加权,加强网络浅层特征的有效复用,提高检测精度;头部网络预测模块引入旋转量,使预测框定位更精准,减少漏检,提高检测准确性;采用相对熵损失函数计算边界框回归损失,解决引入旋转量导致的损失值突变问题;添加第4检测层并聚类分析绝缘子数据集确定预测锚框尺寸,提高网络小目标检测精度;利用剪枝算法压缩模型体积,加快检测速度。实验结果表明:剪枝后的Ghost-YOLOv5s网络检测精度和速度分别为95.9%和78.6 帧/s,满足绝缘子缺陷检测要求。
In order to solve the problem that the background area of insulator detection in the horizontal rectangular frame becomes excessively large due to aerial shooting angles, a rotating bounding box is used instead of the horizontal rectangular frame to detect the insulator, and the Ghost-YOLOv5s network is constructed to achieve high-precision and fast detection of insulator defects using rotating bounding boxes. The lightweight GhostNet is employed as the backbone network of YOLOv5s to reduce network parameters and enhance detection speed. In the neck network, a bidirectional feature pyramid structure is adopted along with the introduction of learnable channel weights for adaptive weighting, strengthening the effective reuse of shallow features in the network and improving detection accuracy. The head network prediction module introduced rotation variables to improve the accuracy of bounding box localization, reduce missed detections, and improve detection accuracy. A relative entropy loss function is used to calculate the bounding box regression loss, addressing the issue of loss value mutation caused by the introduction of rotation variables. A fourth detection layer is added, and clustering analysis on the insulator dataset determine the sizes of the predicted anchor boxes, improving the network’s precision in detecting small targets. A pruning algorithm is applied to compress the model size, further increasing detection speed. Experimental results show that the pruned Ghost-YOLOv5s network achieve a detection precision and speed of 95.9% and 78.6 fps, respectively, meeting the requirements for insulator defect detection.
绝缘子缺陷 / 旋转目标框 / 聚类分析 / 剪枝算法 / 航拍图像 / 轻量化
insulator defects / rotate the target box / cluster analysis / pruning algorithm / aerial image / lightweight
| [1] |
徐建军, 黄立达, 闫丽梅, 等. 基于层次多任务深度学习的绝缘子自爆缺陷检测[J]. 电工技术学报, 2021, 36(7): 1407-1415.
|
| [2] |
|
| [3] |
|
| [4] |
|
| [5] |
|
| [6] |
|
| [7] |
|
| [8] |
|
| [9] |
|
| [10] |
李耀, 胡军国, 乐杨. 融合GhostNet的YOLOv5垃圾分类方法[J]. 电子技术应用, 2024, 50(1): 14-20.
|
| [11] |
朱文超, 杨洁, 何超. 林区视频监控下车流量分类统计 [J].计量学报, 2023, 44(7): 1093-1099.
|
| [12] |
|
| [13] |
胡惠娟, 秦一锋, 徐鹤, 等. 面向无人机航拍图像的YOLOv8目标检测改进算法 [J]. 计算机科学, 2025, 52 (4): 202-211.
|
| [14] |
赵振兵, 蒋志钢, 李延旭, 等. 输电线路部件视觉缺陷检测综述[J]. 中国图象图形学报, 2021, 26 (11): 2545-2560.
|
| [15] |
|
| [16] |
朱奇光, 商健, 刘博, 等.基于无人机航拍视频车辆多目标跟踪算法研究[J]. 计量学报, 2024, 45 (12): 1772-1779.
|
| [17] |
|
| [18] |
|
| [19] |
王媛彬, 韦思雄, 吴华英, 等. 基于改进YOLOv5s的矿井下安全帽佩戴检测算法[J]. 煤炭科学技术, 2025, 53(S1): 366-377.
|
| [20] |
|
| [21] |
|
| [22] |
|
| [23] |
孙宏磊, 陈雯柏, 刘辉翔. 一种改进的YOLOv7‑OBB舰船识别方法[J]. 兵器装备工程学报, 2024, 45(8): 192-198.
|
/
| 〈 |
|
〉 |