王杨杨, 曹晖, 莫文昊. 基于深度学习的改进型YOLOv4输电线路鸟巢检测与识别[J]. 智慧电力, 2023, 51(1): 101-107.
引用本文: 王杨杨, 曹晖, 莫文昊. 基于深度学习的改进型YOLOv4输电线路鸟巢检测与识别[J]. 智慧电力, 2023, 51(1): 101-107.
WANG Yang-yang, CAO Hui, MO Wen-hao. Bird’s Nest Detection and Identification on Improved YOLOv4 Transmission Line Based on Deep Learning[J]. Smart Power, 2023, 51(1): 101-107.
Citation: WANG Yang-yang, CAO Hui, MO Wen-hao. Bird’s Nest Detection and Identification on Improved YOLOv4 Transmission Line Based on Deep Learning[J]. Smart Power, 2023, 51(1): 101-107.

基于深度学习的改进型YOLOv4输电线路鸟巢检测与识别

Bird’s Nest Detection and Identification on Improved YOLOv4 Transmission Line Based on Deep Learning

  • 摘要: 针对输电线路无人机巡视图像经典鸟巢检测算法权重参数范围大、识别效率低、识别精度低的缺点,提出了一种改进型YOLOv4输电线路鸟巢检测与识别方法。首先,选取Mosaic图像增强技术对图片集进行多种变换,变相增加图片集中的小目标数量。其次,在骨干特征提取网络中,通过引入深度可分离卷积来提高检测网络的速度;在YOLO头中,基于K-means++算法改进锚框的大小和比例,基于最小凸集建立回归损失函数。最后,在PANet和YOLO头之间增加2个SPP模块,进一步增强特征融合能力,提高小目标检测能力。利用某供电局无人机巡检图像制作数据集,将提出的算法与其他目标检测算法进行对比实验研究。实验结果表明,改进后的算法有更高的鸟巢检测准确度和更低的运算开销。

     

    Abstract: Aiming at the shortcomings of the classic algorithm of UAV inspection image bird’s nest detection on transmission lines,such as excessive weight parameter scale,low recognition efficiency and recognition accuracy,the paper proposes the bird’s nest detection method for an improved YOLOv4 transmission line. Firstly,a mosaic image enhancement method is used to perform various transformations on a picture set to increase the number of small targets in the picture set. Secondly,the depthwise separable convolution is used in the trunk feature extraction network to improve the speed of the detection network. In the YOLO head,the anchor frame size and proportion are improved based on K-means++ algorithm,and the regression loss function is established based on a minimum convex set. Finally,two SPP modules are added between the PANet and the YOLO head to further improve the feature fusion ability and enhance the detection ability of the small targets. A dataset is provided by using the UAV inspection image of a power supply bureau,and the contrasting experiments of the improved algorithm and other target detection algorithms are carried out. The results show that the improved algorithm has higher bird’s nest detection accuracy and lower computing overhead.

     

/

返回文章
返回