W. Junlong, Kangwei Wei, Z. Wei, Huang Fengbiao, Tao Xuefeng, Wu Qiong
{"title":"基于改进YOLOv5和动态锚盒匹配的头盔检测算法","authors":"W. Junlong, Kangwei Wei, Z. Wei, Huang Fengbiao, Tao Xuefeng, Wu Qiong","doi":"10.1109/ICESIT53460.2021.9696525","DOIUrl":null,"url":null,"abstract":"To solve the problems of low recognition accuracy and undetectable helmet of small targets in helmet detection in complex scenes, a helmet detection algorithm based on improved YOLOv5 and dynamic anchor box matching is proposed to improve the detection efficiency of small helmets in complex scenes. Firstly, by adding a small target detection layer in the YOLOv5 network, the detection accuracy of small targets is preliminarily improved; Secondly, convolution block attention model (CBAM) is added to the feature extraction network to enhance the information transmission between feature layers and the recognition of foreground and background by the neural network; Finally, to further improve the detection rate of small target helmet, the accuracy of a priori frame matching is enhanced by dynamic topK anchor frame matching. The weight of pre-training on the COCO data set is fused for detection and recognition to improve the generalization and accuracy of detection. The experimental results show that in the helmet data set constructed in this paper, the detection accuracy of helmets is 98.2%, and the helmet detection of small targets is realized.","PeriodicalId":164745,"journal":{"name":"2021 IEEE International Conference on Emergency Science and Information Technology (ICESIT)","volume":"33 1","pages":"0"},"PeriodicalIF":0.0000,"publicationDate":"2021-11-22","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"1","resultStr":"{\"title\":\"Helmet Detection Algorithm Based on the Improved YOLOv5 and Dynamic Anchor Box Matching\",\"authors\":\"W. Junlong, Kangwei Wei, Z. Wei, Huang Fengbiao, Tao Xuefeng, Wu Qiong\",\"doi\":\"10.1109/ICESIT53460.2021.9696525\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"To solve the problems of low recognition accuracy and undetectable helmet of small targets in helmet detection in complex scenes, a helmet detection algorithm based on improved YOLOv5 and dynamic anchor box matching is proposed to improve the detection efficiency of small helmets in complex scenes. Firstly, by adding a small target detection layer in the YOLOv5 network, the detection accuracy of small targets is preliminarily improved; Secondly, convolution block attention model (CBAM) is added to the feature extraction network to enhance the information transmission between feature layers and the recognition of foreground and background by the neural network; Finally, to further improve the detection rate of small target helmet, the accuracy of a priori frame matching is enhanced by dynamic topK anchor frame matching. The weight of pre-training on the COCO data set is fused for detection and recognition to improve the generalization and accuracy of detection. The experimental results show that in the helmet data set constructed in this paper, the detection accuracy of helmets is 98.2%, and the helmet detection of small targets is realized.\",\"PeriodicalId\":164745,\"journal\":{\"name\":\"2021 IEEE International Conference on Emergency Science and Information Technology (ICESIT)\",\"volume\":\"33 1\",\"pages\":\"0\"},\"PeriodicalIF\":0.0000,\"publicationDate\":\"2021-11-22\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"1\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"2021 IEEE International Conference on Emergency Science and Information Technology (ICESIT)\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1109/ICESIT53460.2021.9696525\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"\",\"JCRName\":\"\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"2021 IEEE International Conference on Emergency Science and Information Technology (ICESIT)","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1109/ICESIT53460.2021.9696525","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
Helmet Detection Algorithm Based on the Improved YOLOv5 and Dynamic Anchor Box Matching
To solve the problems of low recognition accuracy and undetectable helmet of small targets in helmet detection in complex scenes, a helmet detection algorithm based on improved YOLOv5 and dynamic anchor box matching is proposed to improve the detection efficiency of small helmets in complex scenes. Firstly, by adding a small target detection layer in the YOLOv5 network, the detection accuracy of small targets is preliminarily improved; Secondly, convolution block attention model (CBAM) is added to the feature extraction network to enhance the information transmission between feature layers and the recognition of foreground and background by the neural network; Finally, to further improve the detection rate of small target helmet, the accuracy of a priori frame matching is enhanced by dynamic topK anchor frame matching. The weight of pre-training on the COCO data set is fused for detection and recognition to improve the generalization and accuracy of detection. The experimental results show that in the helmet data set constructed in this paper, the detection accuracy of helmets is 98.2%, and the helmet detection of small targets is realized.