A new method for safety helmet detection based on convolutional neural network

YueJing Qian and Bo Wang

PLOS ONE, 2023, vol. 18, issue 10, 1-16

Abstract: Considering practical issues such as cost control of hardware facilities in engineering projects, it is a challenge to design a robust safety helmet detection method, which can be implemented on mobile or embedded devices with limited computing power. This paper presents an approach to optimize the BottleneckCSP structure in the YOLOv5 backbone network, which can greatly reduce the complexity of the model without changing the size of the network input and output. To eliminate the information loss caused by upsampling and enhance the semantic information of the feature map on the reverse path, this paper designs an upsampling feature enhancement module. Besides, To avoid the negative impact of redundant information generated by feature fusion on the detection results, this paper introduces a self-attention mechanism. That is, using the designed channel attention module and location attention module, adjacent shallow feature maps and upsampled feature maps are adaptively fused to generate new feature maps with strong semantics and precise location information. Compared with the existing methods with the fastest inference speed, under the same compute capability, the proposed method not only has faster inference speed, the FPS can reach 416, but also has better performance with mAP of 94.2%.

Date: 2023
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0292970 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 92970&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0292970

DOI: 10.1371/journal.pone.0292970

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().