关联增强改进的CenterNet安全帽检测方法

doi:10.3778/j.issn.1002-8331.2206-0097

摘要/Abstract

摘要： 施工现场的安全帽佩戴状况自动化检测是保障员工安全的重要手段，目前所面临的挑战包括安全帽目标往往较小、密集且部分被遮挡，难以兼顾方法的精确度和实时性等。为此，提出一种关联增强的CenterNet改进方法。为充分发挥CenterNet逐像素分类的特点，引入关联融合模块来实现深、浅层特征的融合，弥补信息损失；同时使用上下文注意力提升模块来引导关联多级增强，进一步提升检测精度，降低误检率；此外，分阶段实施轻量化策略，剔除冗余、精简网络，极大降低权重规模、提升算法效率。该方法在复杂场景数据集上的准确率为88.6%，平均推理时间12?ms，平均权重大小19.5?MB，均优于主流对比方法。实验结果证明，该方法兼具强实时性与高准确度，适合复杂场景中的安全帽检测。

关键词: 目标检测, CenterNet, 注意力机制, 金字塔池化

Abstract: Automatic helmet wearing detection on construction sites is an important way to ensure the safety of employees. The current challenges include that the helmet targets are often very small, dense and partially blocked, and it is difficult to balance accuracy and real-time, etc. To solve that, an improved CenterNet method with enhanced associations is proposed in this paper. Taking the advantage of the pixel-wise classification feature of the CenterNet, an association module is introduced to realize the fusion of deep and shallow features and compensate for information loss. Besides, a context-attention module is involved to guide the multi-level correlation enhancement, which further improves detection accuracy and reduces false detection rate. In addition, a lightweight strategy is adopted in stages to eliminate redundancy and streamline the network, which greatly reduces the model weights and improves the efficiency of the algorithm. The accuracy of the proposed method on a complex scene public data set is 88.6%, the average forward propagation time is 12?ms, and the average weight size is 19.5 MB, which outperforms the mainstream comparison methods. Experimental results prove that, with the excellent real-time performance and the high accuracy, the proposed method is suitable for helmet wearing detection in complex scenes.

Key words: object detection, CenterNet, attention mechanism, spatial pyramid pooling

黄品超, 刘石坚, 徐戈, 邹峥. 关联增强改进的CenterNet安全帽检测方法[J]. 计算机工程与应用, 2023, 59(17): 250-256.

HUANG Pinchao, LIU Shijian, XU Ge, ZOU Zheng. Helmet Wearing Detection Method Based on Improved CenterNet with Enhanced Associations[J]. Computer Engineering and Applications, 2023, 59(17): 250-256.

参考文献

[1] ZHOU X，WANG D，KR?HENBüHL P.Objectsas points[J].arXiv：1904.07850，2019.
[2] ZHANG H，YAN X，LI H，et al.Real-time alarming，moni-toring，and locating for non-hard-hat use in construction[J].Journal of Computing in Civil Engineering，2019，145（3）：4019006.
[3] MNEYMNEH B E，ABBAS M，KHOURY H.Vision-based framework for intelligent monitoring of hardhat wearing on construction sites[J].Journal of Computing in Civil Engineering，2019，33（2）：1-20.
[4] FANG Q，LI H，LUO X，et al.Detecting non-hardhat-use by a deep learning method from far-field surveillance videos[J].Automation in Construction，2018，85（1）：1-9.
[5] SHEN J，XIONG X，LI Y，et al.Detecting safety helmet wearing on construction sites with bounding-box regression and deep transfer learning[J].Computer-Aided Civil and Infrastructure Engineering，2020，36（2）：180-196.
[6] LI J，WANG Y，WANG C，et al.DSFD：dual shot face detector[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），June 15-20，2019，Long Beach，CA，2019：5060-5069.
[7] 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（CVPR），June 27-30，2016，Las Vegas，NV，USA.New York：IEEE Press，2016.
[8] REDMON J，FARBADI A.YOLO9000：better，faster，stronger[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition（CVPR），July 21-26，2017，Honolulu，HI，USA.New York：IEEE Press，2017：6517-6525.
[9] REDMON J，FARHADI A.YOLOv3：an incremental improvement[J].arXiv：1804.02767，2018.
[10] LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multibox detector[C]//Proceedings of the European Conference on Computer Vision（ECCV），2016：21-37.
[11] 秦伟伟，宋泰年，刘洁瑜，等.基于轻量化YOLOv3的遥感军事目标检测算法[J].计算机工程与应用，2021，57（21）：263-269.
QIN W W，SONG T N，LIU J Y，et at.Remote sensing military target detection algorithm based on lightweight YOLOv3[J].Computer Engineering and Applications，2021，57（21）：263-269.
[12] 王昱潭，朱超伟，赵琛，等.基于Faster R-CNN的灵武长枣图像检测方法[J].计算机工程与应用，2021，57（4）：216-224.
WANG Y T，ZHU C W，ZHAO C，et at.Image detection method of Lingwu long jujube based on Faster R-CNN[J].Computer Engineering and Applications，2021，57（4）：216-224.
[13] 潘慧，段先华，罗斌强.多尺度特征DCA融合的海上船舶检测算法研究[J].计算机工程与应用，2022，58（4）：177-185.
PAN H，DUAN X H，LUO B Q.Research on marine ship detection based on multi-scale feature fusion and DCA[J].Computer Engineering and Applications，2022，58（4）：177-185.
[14] HE K，ZHANG X，REN S，et al.Deep residual learning for image recognition[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition （CVPR），June 27-30，2016，Las Vegas，NV，USA，2016：770-778.
[15] YU F，WANG D，SHELHAMER E，et al.Deep layer aggregation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：2403-2412.
[16] RONNEBERGER O，FISCHER P，BROX T.U-net：convolutional networks for biomedical image segmentation[C]//Proceedings of the International Conference on Medical Image Computing and Computer-Assisted Intervention.Cham：Springer，2015：234-241.
[17] ZHOU Z，SIDDIQUEE M M R，TAJBAKHSH N，et al.Unet++：a nested u-net architecture for medical image segmentation[M]//Deep learning in medical image analysis and multimodal learning for clinical decision support.Cham：Springer，2018：3-11.
[18] HUANG G，LIU Z，LAURENS V，et al.Densely connected convolutional networks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2017：4700-4708.
[19] WU J，CAI N，CHEN W，et al.Automatic detection of hardhats worn by construction personnel：a deep learning approach and benchmark dataset[J].Automation in Construction，2019，106：102894.
[20] SANDLER M，HOWARD A，ZHU M，et al.Mobilenetv2：inverted residuals and linear bottlenecks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：4510-4520.
[21] HOU Q B，ZHOU D Q，FENG J S.Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），2021：13713-13722.