Welcome to Journal of Beijing Institute of Technology
Volume 31Issue 6
Dec. 2022
Turn off MathJax
Article Contents
Yunzuo Zhang, Kaina Guo. Power Plant Indicator Light Detection System Based on Improved YOLOv5[J]. JOURNAL OF BEIJING INSTITUTE OF TECHNOLOGY, 2022, 31(6): 605-612. doi: 10.15918/j.jbit1004-0579.2022.079
Citation: Yunzuo Zhang, Kaina Guo. Power Plant Indicator Light Detection System Based on Improved YOLOv5[J].JOURNAL OF BEIJING INSTITUTE OF TECHNOLOGY, 2022, 31(6): 605-612.doi:10.15918/j.jbit1004-0579.2022.079

Power Plant Indicator Light Detection System Based on Improved YOLOv5

doi:10.15918/j.jbit1004-0579.2022.079
Funds:This work was supported by the National Natural Science Foundation of China (Nos. 61702347, 62027801), the Natural Science Foundation of Hebei Province (Nos. F2022210007, F2017210161), the Science and Technology Project of Hebei Education Department (Nos. ZD2022100, QN2017132), and the Central Guidance on Local Science and Technology Development Fund (No.226Z0501G).
More Information
  • Author Bio:

    Yunzuo Zhangreceived the B.S. degree from School of Information Science and Engineering, Hebei University of Science and Technology, Hebei in 2007, the M.S. degree from School of Information and Communication, Guilin University of Electronic Science and Technology, Guilin, in 2011 and the Ph.D. degree from School of Information and Electronics, Beijing Institute of Technology, Beijing, in 2016. In 2018, he was a visiting scholar in California State University. He is currently a Professor with School of Information Science and Technology, Shijiazhuang Tiedao University, Hebei, China. His research interests include video processing, image processing and radar signal processing

    Kaina Guoreceived the B.S. degree in software engineering from the Shijiazhuang University of Computer Science and Engineering, in 2018. She is currently a graduate research candidate with the Computer Science and Technology of Shijiazhuang Tiedao University. Her main interests include image processing, surveillance video analysis and processing

  • Corresponding author:zhangyunzuo888@sina.com
  • Received Date:2022-07-15
  • Rev Recd Date:2022-09-05
  • Accepted Date:2022-09-10
  • Publish Date:2022-12-25
  • Electricity plays a vital role in daily life and economic development. The status of the indicator lights of the power plant needs to be checked regularly to ensure the normal supply of electricity. Aiming at the problem of a large amount of data and different sizes of indicator light detection, we propose an improved You Only Look Once vision 5 (YOLOv5) power plant indicator light detection algorithm. The algorithm improves the feature extraction ability based on YOLOv5s. First, our algorithm enhances the ability of the network to perceive small objects by combining attention modules for multi-scale feature extraction. Second, we adjust the loss function to ensure the stability of the object frame during the regression process and improve the convergence accuracy. Finally, transfer learning is used to augment the dataset to improve the robustness of the algorithm. The experimental results show that the average accuracy of the proposed squeeze-and-excitation YOLOv5s (SE-YOLOv5s) algorithm is increased by 4.39% to 95.31% compared with the YOLOv5s algorithm. The proposed algorithm can better meet the engineering needs of power plant indicator light detection.
  • loading
  • [1]
    X. Yan and H. You, “Object detection algorithm for faster-RCNN power transmission tower, ” Computer Simulation, vol. 37, no. 2, pp. 135-139, 2020.
    [2]
    C. Wu, G. Ma, S. Yang, Y. Wu, and Y. Kong, “Virtual power plant operation optimization strategy based on analysis object cascading of distributed transaction module, ” Power Demand Side Management, vol. 24, no. 1, pp. 7-13, 2022.
    [3]
    D. Peng, W. Liu, E. Qi, and J. Hu, “Research on leakage detection of high pressure steam in power plant based on CBAM-Res_Unet, ” Journal of Electronic Measurement and Instrumentation, vol. 35, no. 12, pp. 206-214, 2021.
    [4]
    P. Yang, “Analysis of countermeasures for stable operation of power plant automation control system, ” Application of IC, vol. 39, no. 6, pp. 274-276, 2022.
    [5]
    H. Chen, H. Zhang, and Z. Wang, “Mode of virtual power plant operation and dispatching in energy blockcnan network, ” Journal of Systems& Management, vol. 31, no. 1, pp. 143-149, 2022.
    [6]
    Y. Zhang, P. Yang, and W. Li, “Detection of leakage cable fixture in tunnel based on feature enhancement SSD, ” Journal of Beijing Jiaotong University, vol. 45, no. 2, pp. 105-110, 2021.
    [7]
    L. Fang, H. Hang, and G. Zhou, “Overview of object detection algorithms, ” Computer Engineering and Applications, vol. 43, no. 13, pp. 11-18, 2018.
    [8]
    M. Meng, X. Gu, Y. Liang, D. Xu, and C. Wu, “Deep neural architecture search: A survey, ” Journal of Computer Research and Development, vol. 58, no. 1, pp. 22-23, 2021.
    [9]
    Z. Shen, Z. Liu, J. Li, Y. Jiang, Y. Chen, and X. Xue, “Object detection from scratch with deep supervision, ” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 42, no. 2, pp. 398-412, 2020.
    [10]
    R. Girshick, “Fast R-CNN, ” in IEEE International Conference on Computer Vision, Santiago, Chile, pp. 1440-1448, 2015.
    [11]
    S. Ren, K. He, R. Girshick, and J. Sun, “Faster R-CNN: Towards real-time object detection with region proposal networks, ” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 39, no. 6, pp. 1137-1149, 2017.
    [12]
    W. Liu, D. Anguelov, D. Erhan, C. Szegedy, S Reed, C. Y. Fu, and A. C. Berg, “SSD: Single shot multibox detector, ” in European Conference on Computer Vision, Amsterdam, Netherlands, pp. 21-37, 2016.
    [13]
    L. Zhu, X. Geng, Li Z, and C. Liu, “Improving YOLOv5 with attention mechanism for detecting boulders from planetary images, ” Remote Sensing, vol. 13, no. 18, pp. 3776, 2021.
    [14]
    W. Huang, R. Chen, and T. Yuan, “Compression of UAV object detection module based on improved YOLOv3-SPP, ” Computer Engineering and Applications, vol. 57, no. 21, pp. 165-173, 2021.
    [15]
    J. Redmon, S. Divvala, R. Girshick, and A. Farhadi, “You only look once: Unified, real-time object detection, ” in IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, pp. 779-788, 2016.
    [16]
    J. Redmon and A. Farhadi, “YOLO9000: Better, faster, stronger, ” in IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA, pp. 6517-6525, 2017.
    [17]
    J. Redmon and A. Farhadi, “Yolov3: An incremental improvement, ” CoRR, arXiv: 1804.02767, 2018.
    [18]
    A. Bochkovskiy, C. Wang, and H. M. Liao, “Yolov4: Optimal speed and accuracy of object detection, ” CoRR, arXiv: 2004.10934, 2020.
    [19]
    P. Ren, L. Wang, W. Fang, S. Song, and S. Djahel, “A novel squeeze yolo-based real-time people counting approach, ” International Journal of Bio-Inspired Computation, vol. 16, no. 2, pp. 94-101, 2020.
    [20]
    J. Hu, L. Shen, and G. Sun, “Squeeze-and-excitation networks” in IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA, pp. 7132-7141, 2018.
    [21]
    L. Guo, Q. Wang, W. Xue, and J. Guo, “A small object detection algorithm based on improved YOLOv5, ” Journal of University of Electronic Science and Technology of China, vol. 51, no. 2, pp. 251-258, 2022.
    [22]
    Z. Zheng, P. Wang, W. Liu, J. Li, R. Ye, and D. Ren, “Distance-IoU loss: Faster and better learning for bounding box regression, ” in AAAI Conference on Artificial Intelligence, New York, USA, vol. 34, no. 07, pp. 12993-13000, 2020.
  • 加载中

Catalog

    通讯作者:陈斌, bchen63@163.com
    • 1.

      沈阳化工大学材料科学与工程学院 沈阳 110142

    1. 本站搜索
    2. 百度学术搜索
    3. 万方数据库搜索
    4. CNKI搜索

    Figures(9)/Tables(1)

    Article Metrics

    Article views (69) PDF downloads(14) Cited by()
    Proportional views
    Related

    /

    Return
    Return
      Baidu
      map