Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection
Safety production management is an important policy for the development of high-risk enterprises such as the construction industry and heavy industry,and safety helmets play a key role in head protection in the production environment.Therefore,it is necessary to strengthen the supervision of helmet...
| Main Author: | |
|---|---|
| Format: | Article |
| Language: | zho |
| Published: |
Editorial office of Computer Science
2021-11-01
|
| Series: | Jisuanji kexue |
| Subjects: | |
| Online Access: | https://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-11-268.pdf |
| _version_ | 1818338233069600768 |
|---|---|
| author | JIN Yu-fang, WU Xiang, DONG Hui, YU Li, ZHANG Wen-an |
| author_facet | JIN Yu-fang, WU Xiang, DONG Hui, YU Li, ZHANG Wen-an |
| author_sort | JIN Yu-fang, WU Xiang, DONG Hui, YU Li, ZHANG Wen-an |
| collection | DOAJ |
| description | Safety production management is an important policy for the development of high-risk enterprises such as the construction industry and heavy industry,and safety helmets play a key role in head protection in the production environment.Therefore,it is necessary to strengthen the supervision of helmet wearing.In recent years,the monitoring method of helmet wearing based on image vision has become the main means for enterprises to implement management.How to improve the detection accuracy and speed of helmet wearing is a crucial issue for applications.To deal with this issue,an improved YOLO v4 algorithm is proposed to promote the accuracy and efficiency of safety helmet wearing detection in this paper.First,a 128×128 feature map output is added to the original three feature map outputs of the YOLO v4 algorithm,and the 8 times downsampling of the feature map output is changed to 4 times downsampling to provide more small target features for subsequent feature fusion.Second,the feature fusion module is improved based on the idea of dense connection to realize feature reuse,so that the Yolo Head classifier responsible for small target detection can utilize the features of different levels,to obtain better target detection and classification results.Finally,comparative experiments are carried out.The results show that the average accuracy of the proposed method is 2.96% higher than the original network detection accuracy to be 91.17%,and the detection speed is basically unchanged to be 52.9 frame/s.Thereupon,the proposed algorithm can achieve better detection accuracy while meeting real-time detection requirements,and effectively realize the high-speed and high-precision detection of helmet wearing. |
| first_indexed | 2024-12-13T15:07:51Z |
| format | Article |
| id | doaj.art-3da3749119244018b4780d4d00289302 |
| institution | Directory Open Access Journal |
| issn | 1002-137X |
| language | zho |
| last_indexed | 2024-12-13T15:07:51Z |
| publishDate | 2021-11-01 |
| publisher | Editorial office of Computer Science |
| record_format | Article |
| series | Jisuanji kexue |
| spelling | doaj.art-3da3749119244018b4780d4d002893022022-12-21T23:40:58ZzhoEditorial office of Computer ScienceJisuanji kexue1002-137X2021-11-01481126827510.11896/jsjkx.200900098Improved YOLO v4 Algorithm for Safety Helmet Wearing DetectionJIN Yu-fang, WU Xiang, DONG Hui, YU Li, ZHANG Wen-an0College of Information Engineering,Zhejiang University of Technology,Hangzhou 310023,China<br/>United Key Laboratory of Embedded System of Zhejiang Province,Hangzhou 310023,ChinaSafety production management is an important policy for the development of high-risk enterprises such as the construction industry and heavy industry,and safety helmets play a key role in head protection in the production environment.Therefore,it is necessary to strengthen the supervision of helmet wearing.In recent years,the monitoring method of helmet wearing based on image vision has become the main means for enterprises to implement management.How to improve the detection accuracy and speed of helmet wearing is a crucial issue for applications.To deal with this issue,an improved YOLO v4 algorithm is proposed to promote the accuracy and efficiency of safety helmet wearing detection in this paper.First,a 128×128 feature map output is added to the original three feature map outputs of the YOLO v4 algorithm,and the 8 times downsampling of the feature map output is changed to 4 times downsampling to provide more small target features for subsequent feature fusion.Second,the feature fusion module is improved based on the idea of dense connection to realize feature reuse,so that the Yolo Head classifier responsible for small target detection can utilize the features of different levels,to obtain better target detection and classification results.Finally,comparative experiments are carried out.The results show that the average accuracy of the proposed method is 2.96% higher than the original network detection accuracy to be 91.17%,and the detection speed is basically unchanged to be 52.9 frame/s.Thereupon,the proposed algorithm can achieve better detection accuracy while meeting real-time detection requirements,and effectively realize the high-speed and high-precision detection of helmet wearing.https://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-11-268.pdfdeep learning|small target detection|safety helmet wearing detection|yolo v4|k-means clustering |
| spellingShingle | JIN Yu-fang, WU Xiang, DONG Hui, YU Li, ZHANG Wen-an Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection Jisuanji kexue deep learning|small target detection|safety helmet wearing detection|yolo v4|k-means clustering |
| title | Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection |
| title_full | Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection |
| title_fullStr | Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection |
| title_full_unstemmed | Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection |
| title_short | Improved YOLO v4 Algorithm for Safety Helmet Wearing Detection |
| title_sort | improved yolo v4 algorithm for safety helmet wearing detection |
| topic | deep learning|small target detection|safety helmet wearing detection|yolo v4|k-means clustering |
| url | https://www.jsjkx.com/fileup/1002-137X/PDF/1002-137X-2021-11-268.pdf |
| work_keys_str_mv | AT jinyufangwuxiangdonghuiyulizhangwenan improvedyolov4algorithmforsafetyhelmetwearingdetection |