Construction personnel dress code detection based on YOLO framework  

作　　者：Yunkai Lyu Xiaobing Yang Ai Guan Jingwen Wang Leni Dai 

机构地区：[1]College of Information Engineering,China Jiliang University,Hangzhou,China [2]College of Integrated Circuit Science and Engineering,Nanjing University of Posts and Telecommunications,Nanjing,China [3]China Jiliang University,Hangzhou,Zhejiang,China [4]Department of Mathematics,University of California San Diego,La Jolla,California,USA

出　　处：《CAAI Transactions on Intelligence Technology》2024年第3期709-721,共13页智能技术学报（英文）

摘　　要：It is important for construction personnel to observe the dress code,such as the correct wearing of safety helmets and reflective vests is conducive to protecting the workers'lives and safety of construction.A YOLO network-based detection algorithm is proposed for the construction personnel dress code(YOLO-CPDC).Firstly,Multi-Head Self-Attention(MHSA)is introduced into the backbone network to build a hybrid back-bone,called Convolution MHSA Network(CMNet).The CMNet gives the model a global field of view and enhances the detection capability of the model for small and obscured targets.Secondly,an efficient and lightweight convolution module is designed.It is named Ghost Shuffle Attention-Conv-BN-SiLU(GSA-CBS)and is used in the neck network.The GSANeck network reduces the model size without affecting the perfor-mance.Finally,the SIoU is used in the loss function and Soft NMS is used for post-processing.Experimental results on the self-constructed dataset show that YOLO-CPDC algorithm has higher detection accuracy than current methods.YOLO-CPDC achieves a mAP50 of 93.6%.Compared with the YOLOv5s,the number of parame-ters of our model is reduced by 18%and the mAP50 is improved by 1.1%.Overall,this research effectively meets the actual demand of dress code detection in construction scenes.

关 键 词：artificial intelligence computer vision deep learning deep neural networks 

分 类 号：TP3[自动化与计算机技术—计算机科学与技术] TP18