Suppr超能文献

绝缘子目标检测与缺陷识别的轻量级算法

A Lightweight Algorithm for Insulator Target Detection and Defect Identification.

机构信息

Department of Electronic and Electrical Engineering, Wuhan Textile University, Wuhan 430200, China.

State Key Laboratory of New Textile Materials and Advanced Processing Technologies, Wuhan Textile University, Wuhan 430200, China.

出版信息

Sensors (Basel). 2023 Jan 20;23(3):1216. doi: 10.3390/s23031216.

Abstract

The accuracy of insulators and their defect identification by UAVs (unmanned aerial vehicles) in transmission-line inspection needs to be further improved, and the model size of the detection algorithm is significantly reduced to make it more suitable for edge-end deployment. In this paper, the algorithm uses a lightweight GhostNet module to reconstruct the backbone feature extraction network of the YOLOv4 model and employs depthwise separable convolution in the feature fusion layer. The model is lighter on the premise of ensuring the effect of image information extraction. Meanwhile, the ECA-Net channel attention mechanism is embedded into the feature extraction layer and PANet (Path Aggregation Network) to improve the recognition accuracy of the model for small targets. The experimental results show that the size of the improved model is reduced from 244 MB to 42 MB, which is only 17.3% of the original model. At the same time, the mAp of the improved model is 0.77% higher than that of the original model, reaching 95.4%. Moreover, the mAP compared with YOLOv5-s and YOLOX-s, respectively, is improved by 1.98% and 1.29%. Finally, the improved model is deployed into Jetson Xavier NX and run at a speed of 8.8 FPS, which is 4.3 FPS faster than the original model.

摘要

无人机 (UAV) 在输电线路巡检中对绝缘子的准确性及其缺陷识别有待进一步提高,并且检测算法的模型大小显著减小,使其更适合边缘端部署。本文算法采用轻量级 GhostNet 模块对 YOLOv4 模型的骨干特征提取网络进行重构,并在特征融合层采用深度可分离卷积。在保证图像信息提取效果的前提下,使模型更轻量。同时,将 ECA-Net 通道注意力机制嵌入到特征提取层和 PANet(Path Aggregation Network)中,以提高模型对小目标的识别精度。实验结果表明,改进后的模型大小从 244MB 减小到 42MB,仅为原始模型的 17.3%。同时,改进后的模型的 mAP 比原始模型高 0.77%,达到 95.4%。此外,与 YOLOv5-s 和 YOLOX-s 相比,分别提高了 1.98%和 1.29%。最后,将改进后的模型部署到 Jetson Xavier NX 中,运行速度为 8.8FPS,比原始模型快 4.3FPS。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/5cc8/9921440/a2db67782fd5/sensors-23-01216-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验