Abstract
Currently, the use of deep learning technologies for detecting defects in transmission line insulators based on images obtained through unmanned aerial vehicle inspection simultaneously presents the problems of insufficient detection accuracy and speed. Therefore, this study first introduced the bidirectional feature pyramid network (BiFPN) module into YOLOv5 to achieve high detection speed as well as enable the combination of image features at different scales, enhance information representation, and allow accurate detection of insulator defect at different scales. Subsequently, the BiFPN module and simple parameter-free attention module (SimAM) were combined to improve the feature representation ability and object detection accuracy. The SimAM also enabled fusion of features at multiple scales, further improving the insulator defect detection performance. Finally, multiple experimental controls were designed to verify the effectiveness and efficiency of the proposed model. The experimental results obtained using self-made datasets show that the combined BiFPN and SimAM model (i.e., the improved BaS-YOLOv5 model) performs better than the original YOLOv5 model; the precision, recall, average precision and F1 score increased by 6.2%, 5%, 5.9%, and 6%, respectively. Therefore, BaS-YOLOv5 substantially improves detection accuracy while maintaining a high detection speed, meeting the requirements for real-time insulator defect detection.
This is a preview of subscription content, log in via an institution to check access.
Similar content being viewed by others
Data availability
All data generated or analyzed during this study are included in this published article.
References
Li, Y., Ni, M., Lu, Y.: Insulator defect detection for power grid based on light correction enhancement and YOLOv5 model. Energy Rep. 8, 807–814 (2022). https://doi.org/10.1016/j.egyr.2022.08.027
Li, J., Yang, Y.: HM-YOLOv5: A fast and accurate network for defect detection of hot-pressed light guide plates. Eng. Appl. Artif. Intell. 117, 105529 (2023). https://doi.org/10.1016/j.engappai.2022.105529
Mahendrakar, T., Ekblad, A., Fischer, N., White, R., Wilde, M., Kish, B., Silver, I.: Performance study of YOLOv5 and faster R-CNN for autonomous navigation around non-cooperative targets. 2022 IEEE Aerosp. Conf. 1–12 (2022). https://doi.org/10.1109/AERO53065.2022.9843537
Tang, J., Liu, S., Zhao, D., Tang, L., Zou, W., Zheng, B.: PCB-YOLO: An improved detection algorithm of PCB surface defects based on YOLOv5. Sustainability. 15, 5963 (2023). https://doi.org/10.3390/su15075963
Jia, Z., Su, X., Ma, G., Dai, T., Sun, J.: Crack identification for marine engineering equipment based on improved SSD and YOLOv5. Ocean. Eng. 268, 113534 (2023). https://doi.org/10.1016/j.oceaneng.2022.113534
Jiao, L., Zhang, F., Liu, F., Yang, S., Li, L., Feng, Z., Qu, R.: A survey of deep learning-based object detection. IEEE Access. 7, 128837–128868 (2019). https://doi.org/10.1109/ACCESS.2019.2939201
Bi, X., Hu, J., Xiao, B., Li, W., Gao, X.: IEMask R-CNN: Information-enhanced mask R-CNN. IEEE Trans. Big Data. 9, 688–700 (2022). https://doi.org/10.1109/TBDATA.2022.3187413
Zhang, Y., Guo, Z., Wu, J., Tian, Y., Tang, H., Guo, X.: Real-time vehicle detection based on improved YOLO v5. Sustainability. 14, 12274 (2022). https://doi.org/10.3390/su141912274
Zou, X., Chen, Z., Zhou, Y.: A lightweight model based on YOLOv5 for helmet wearing detection. 4th International Conference on Informatics Engineering and Information Science (ICIEIS2021) 12161:121610Z. (2022). https://doi.org/10.1117/12.2627279
Tan, M., Pang, R., Le, Q.V.: Efficientdet: scalable and efficient object detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 10778–10787 (2020)
Yang, L., Zhang, R.Y., Li, L.: Simam: a simple, parameter-free attention module for convolutional neural networks. International Conference on Machine Learning 11863–11874 (2021)
Wu, D., Lv, S., Jiang, M., Song, H.: Using channel pruning-based YOLO v4 deep learning algorithm for the real-time and accurate detection of apple flowers in natural environments. Comput. Electron. Agric. 178, 105742 (2020). https://doi.org/10.1016/j.compag.2020.105742
Kim, J.-H., Kim, N., Park, Y.W., Won, C.S.: Object detection and classification based on YOLO-V5 with improved maritime dataset. J. Mar. Sci. Eng. 10, 377 (2022). https://doi.org/10.3390/jmse10030377
Wan, G., Fang, H., Wang, D., Yan, J., Xie, B.: Ceramic tile surface defect detection based on deep learning. Ceram. Int. 48, 11085–11093 (2022). https://doi.org/10.1016/j.ceramint.2021.12.328
Zhang, M., Yin, L.: Solar cell surface defect detection based on improved YOLO v5. IEEE Access. 10, 80804–80815 (2022). https://doi.org/10.1109/ACCESS.2022.3195901
Mushtaq, F., Ramesh, K., Deshmukh, S., Ray, T., Parimi, C., Tandon, P., Jha, P.K.: Nuts&bolts: YOLO-v5 and image processing based component identification system. Eng. Appl. Artif. Intell. 118, 105665 (2023). https://doi.org/10.1016/j.engappai.2022.105665
Ghiasi, G., Lin, T.Y., Le, Q.V.: Nas-fpn: learning scalable feature pyramid architecture for object detection. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition 7029–7038 (2019)
Wang, X., Wu, Z., Jia, M., Xu, T., Pan, C., Qi, X., Zhao, M.: Lightweight SM-YOLOv5 tomato fruit detection algorithm for plant factory. Sensors. 23, 3336 (2023). https://doi.org/10.3390/s23063336
Wu, B., Yuan, S., Li, P., Jing, Z., Huang, S., Zhao, Y.: Radar emitter signal recognition based on one-dimensional convolutional neural network with attention mechanism. Sensors. 20, 6350 (2020). https://doi.org/10.3390/s20216350
Cai, S., Wang, C., Ding, J., Yu, J., Fan, J.: FDAM: Full-dimension attention module for deep convolutional neural networks. Int. J. Multimedia Inf. Retr. 11, 599–610 (2022). https://doi.org/10.1007/s13735-022-00248-3
Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.-Y., Berg, A.C.: SSD: Single shot multibox detector. Comput. vision–ECCV 2016. 9905 (2016). https://doi.org/10.1007/978-3-319-46448-0_2
Girshick, R.: Fast R-CNN. Proceedings of the IEEE International Conference on Computer Vision 1440–1448 (2015)
Redmon, J., Farhadi, A.: YOLOv3: An incremental improvement. ArXiv e-Prints. (2018). https://doi.org/10.48550/arXiv.1804.02767
Wang, C.Y., Bochkovskiy, A., Liao, H.-Y.M.: YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. ArXiv e-Prints. (2022). https://doi.org/10.48550/arXiv.2207.02696
Raimundo, A.: Insulator data set - Chinese power line insulator dataset (CPLID). IEEE Dataport. (2020). https://doi.org/10.21227/qtxb-2s61
Zhang, Z.D., Zhang, B., Lan, Z.C., Liu, H.C., Li, D.Y., Pei, L., Yu, W.X.: FINet: An insulator dataset and detection benchmark based on synthetic fog and improved YOLOv5. IEEE Trans. Instrum. Meas. 71, 1–8 (2022). https://doi.org/10.1109/TIM.2022.3194909
Acknowledgements
This work was supported by the Shanxi Provincial Higher Education Science and Technology Innovation Project (Grant number 2022L524) and Shanxi Provincial Key Research and Development Project (Grant Number 202102060301020).
Funding
There was no specific funding for this study.
Author information
Authors and Affiliations
Contributions
Y. Z. wrote the main manuscript text. Y. Z. and Y.D. designed the experiment；K.Y. and X.S.prepared all figures. J.W. and L.Z.prepared all tables. All authors reviewed the manuscript.
Corresponding author
Ethics declarations
Competing interests
The authors declare no competing interests.
Additional information
Communicated by An-An Liu.
Publisher’s Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
