Journals
  Publication Years
  Keywords
Search within results Open Search
Please wait a minute...
For Selected: Toggle Thumbnails
Lightweight safety helmet wearing detection algorithm based on YOLO v8
Yong FENG, Sizhuo YANG, Hongyan XU
Journal of Computer Applications    0, (): 251-256.   DOI: 10.11772/j.issn.1001-9081.2024010020
Abstract249)   HTML0)    PDF (2164KB)(665)    PDF(mobile) (1480KB)(22)    Save

Construction, mining, exploration and other industries have mandatory regulations on helmet wearing in production. The helmet wearing detection algorithms have been widely used in the above industries, but the existing algorithms have problems such as too many parameters, high complexity and poor real-time performance. Therefore, a lightweight safety helmet wearing detection algorithm YOLO v8-s-LE was proposed on the basis of YOLO v8 (You Only Look Once v8). Firstly, the LAD (Light Adaptive-weight Downsampling) method was designed,so that compared with the original YOLO v8 algorithm, the proposed algorithm reduced floating-point computation significantly. Then, the efficient multi-scale convolution C2f_EMC (C2f_Efficient Multi-Scale Conv) method was used to extract multi-scale feature information, which increased the depth of the network effectively, made the neural network take into account both shallow and deep semantic information, and further improved the expression ability of the algorithm for feature information. Experimental results show that compared with YOLO v8-s algorithm on the public dataset SHWD (Safety Helmet Wearing Dataset), the proposed algorithm has the parameters reduced by 77%, the floating-point computation reduced by 73%, and the precision reached 92.6%, verifying that the algorithm takes into account the requirements of accuracy and real-time performance, and is more suitable for the deployment and application in actual production environments.

Table and Figures | Reference | Related Articles | Metrics