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.
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。