Computer Engineering and Applications ›› 2024, Vol. 60 ›› Issue (14): 319-328.DOI: 10.3778/j.issn.1002-8331.2311-0160

• Engineering and Applications • Previous Articles     Next Articles

YOLOv7-tiny Transmission Line Foreign Object Detection Algorithm Based on Channel Pruning

SUN Yang, LI Jia   

  1. School of Information and Control Engineering, Jilin Institute of Chemical Technology, Jilin 132022, China
  • Online:2024-07-15 Published:2024-07-15

基于通道剪枝的YOLOv7-tiny输电线路异物检测算法

孙阳,李佳   

  1. 吉林化工学院 信息与控制工程学院,吉林 132022

Abstract: In response to the problem of poor accuracy and the large model size in transmission line foreign object detection, an improved YOLOv7-tiny algorithm based on channel pruning has been proposed. Firstly, the ReXNet network is used to replace the backbone network of YOLOv7-tiny to address the feature bottleneck issue in the original network. Secondly, diversified branch blocks are introduced to enhance the network’s feature fusion capability. Finally, through layer-adaptive magnitude-based pruning (LAMP), a pruning approach is employed to trade off some accuracy for a reduction in model size and computational load, preparing it for deployment on embedded devices. Experimental results indicate that the final improved model achieves a 3?percentage points increase in accuracy compared to the YOLOv7-tiny model, a 119.4% increase in FPS, and compresses the model size to 14% of the original size.

Key words: transmission lines, YOLOv7-tiny algorithm, channel pruning, foreign object detection

摘要: 针对输电线路异物检测精度不佳且模型庞大的问题,提出了基于通道剪枝的改进YOLOv7-tiny算法用于输电线路异物检测。用ReXNet网络替代了YOLOv7-tiny的骨干网络改进原网络的特征瓶颈问题。引入了多样化分支块从而增加网络特征融合能力,通过基于层自适应幅度的修剪(LAMP)剪枝方案损失一定精度换取模型体积、运算量的降低,为下一步部署到嵌入式设备做好准备。实验结果表明,最终的改进模型相对于YOLOv7-tiny模型精度上提升3个百分点,FPS提升原来的119.4%,模型大小压缩到原来的14%。

关键词: 输电线路, YOLOv7-tiny算法, 通道剪枝, 异物检测