改进目标检测算法在变电站内安全管控的应用

doi:10.3778/j.issn.1002-8331.2107-0005

摘要/Abstract

摘要： 变电站作为电力系统中重要一环，保证变电站作业人员的安全是至关重要的。为了自动检测作业人员是否正确穿着工作服佩戴安全帽，提出一种基于Transformer自注意力编码特征融合轻量级的目标检测网络。通过采用轻量级的主干网络提取特征，注意力机制融合多尺度特征。提出了质量焦点损失方法，改善目标检测模型训练和测试阶段推理过程不一致问题。同时，采集并标注变电站场景下人员工作服安全帽数据5?200张。将提出的轻量级目标模型在自制的工作服安全帽数据集上训练，并在测试集上验证，该目标检测方法识别mAP达44.6%，AP50达79.5%，达到117 FPS。

关键词: 目标检测, 深度学习, 轻量化, Transformer, 神经网络

Abstract: As an important part of the power system, it is crucial to ensure the safety of substation operators. In order to automatically detect whether the operators are correctly wearing overalls and helmets, a Transformer-based self-attentive coding feature fusion lightweight target detection network is proposed. The attention mechanism fuses multi-scale features by using a lightweight backbone network to extract features. A quality focus loss method is proposed to improve the inconsistent inference process in the training and testing phases of the target detection model. Meanwhile, 5,200 pieces of personnel workwear helmet data are collected and labeled in the substation scene. The proposed lightweight target model is trained on the homemade workwear helmet dataset and validated on the test set, and the target detection method recognizes mAP up to 44.6% and AP50 up to 79.5%, reaching 117 FPS.

Key words: object detection, deep learning, light weight, Transformer, neural network

吴宏毅, 雷景生, 陈林锋, 杨胜英. 改进目标检测算法在变电站内安全管控的应用[J]. 计算机工程与应用, 2022, 58(24): 313-320.

WU Hongyi, LEI Jingsheng, CHEN Linfeng, YANG Shengying. Improved Detection Algorithm and Its Application in Safety Control in Substation Scenario[J]. Computer Engineering and Applications, 2022, 58(24): 313-320.

参考文献

[1] 王兵，李文璟，唐欢.改进YOLO v3算法及其在安全帽检测中的应用[J].计算机工程与应用，2020，56（9）：33-40.
WANG B，LI W J，TANG H.Improved YOLO v3 algorithm and its application in helmet detection[J].Computer Engineering and Applications，2020，56（9）：33-40.
[2] FENG G，CHEN Y，CHEN N，et al.Research on automatic identification technology of the safety helmet based on machine vision[J].Machine Design and Manufacturing Engineering，2015，44（10）：39-42.
[3] DAHIYA K，SINGH D，MOHAN C K.Automatic detection of bike-riders without helmet using surveillance videos in real-time[C]//2016 International Joint Conference on Neural Networks，2016：3046-3051.
[4] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，2015：1440-1448.
[5] GIRSHICK R，DONAHUE J，DARRELL T，et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]//2014 IEEE Conference on Computer Vision and Pattern Recognition，2014：580-587.
[6] REN S，HE K，GIRSHICK R，et al.Faster R-CNN：towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2017，39（6）：1137-1149.
[7] HE K，GKIOXARI G，DOLLáR P，et al.Mask R-CNN[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2017，42（2）：386-397.
[8] REDMON J，DIVVALA S，GIRSHICK R，et al.You only look once：unified，real-time object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2016：779-788.
[9] REDMON J，FARHADI A.YOLO9000：better，faster，stronger[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition，2017：6517-6525.
[10] REDMON J，FARHADI A.YOLOv3：an incremental improvement[J].arXiv：1804.02767，2018.
[11] LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multiBox detector[C]//European Conference on Computer Vision，2016.
[12] WANG H，HU Z，GUO Y，et al.A real-time safety helmet wearing detection approach based on CSYOLOv3[J].Applied Sciences，2020，10（19）：6732.
[13] LI Y，WEI H，HAN Z，et al.Deep learning-based safety helmet detection in engineering management based on convolutional neural networks[J].Advances in Civil Engineering，2020.
[14] VASWANI A，SHAZEER N，PARMAR N，et al.Attention is all you need[J].arXiv：1706.03762，2017.
[15] TIAN Z，SHEN C，CHEN H，et al.Fcos：fully convolutional one-stage object detection[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision，2019：9627-9636.
[16] ZHANG S，CHI C，YAO Y，et al.Bridging the gap between anchor-based and anchor-free detection via adaptive training sample selection[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2020：9759-9768.
[17] 邹承明，薛榕刚.融合GIoU和Focal loss的YOLOv3目标检测算法[J].计算机工程与应用，2020，56（24）：214-222.
ZOU C M，XUE R G.Improved YOLOv3 object detection algorithm：combining GIoU and Focal loss[J].Computer Engineering and Applications，2020，56（24）：214-222.
[18] MA N，ZHANG X，ZHENG H T，et al.Shufflenet v2：practical guidelines for efficient cnn architecture design[C]//Proceedings of the European Conference on Computer Vision（ECCV），2018：116-131.
[19] DENG J，DONG W，SOCHER R，et al.Imagenet：a large-scale hierarchical image database[C]//2009 IEEE Conference on Computer Vision and Pattern Recognition，2009：248-255.
[20] LIN T Y，GOYAL P，GIRSHICK R，et al.Focal loss for dense object detection[C]//Proceedings of the IEEE International Conference on Computer Vision，2017：2980-2988.
[21] CHOLLET F.Xception：deep learning with depthwise separable convolutions[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：1251-1258.
[22] HOWARD A G，ZHU M，CHEN B，et al.Mobilenets：efficient convolutional neural networks for mobile vision applications[J].arXiv：1704.04861，2017.
[23] ZHANG X，ZHOU X，LIN M，et al.Shufflenet：an extremely efficient convolutional neural network for mobile devices[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：6848-6856.
[24] LIN T Y，DOLLáR P，GIRSHICK R，et al.Feature pyramid networks for object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：2117-2125.
[25] LIU S，QI L，QIN H，et al.Path aggregation network for instance segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：8759-8768.
[26] LIN D，SHEN D，SHEN S，et al.ZigZagNet：fusing top-down and bottom-up context for object segmentation[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），2019.
[27] WU S，YANG J，WANG X，et al.IoU-balanced loss functions for single-stage object detection[J].arXiv：1908.
05641，2019.
[28] WU S，LI X，WANG X.IoU-aware single-stage object detector for accurate localization[J].Image and Vision Computing，2020，97：103911.
[29] QIN Z，LI Z，ZHANG Z，et al.Thundernet：towards real-time generic object detection on mobile devices[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision，2019：6718-6727.
[30] WANG C Y，LIAO H Y M，WU Y H，et al.CSPNet：a new backbone that can enhance learning capability of CNN[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops，2020：390-391.
[31] WANG C Y，BOCHKOVSKIY A，LIAO H Y M.Scaled-yolov4：scaling cross stage partial network[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2021：13029-13038.