Helmet wearing detection method based on improved YOLOv4 neck optimized network
CSTR:
Author:
Affiliation:

School of Energy & Electronic Engineering, Chang’an University, Xi’an 710016, P. R.China

Fund Project:

Supported by the Key Research and Development Program of Shaanxi Province (2021GY-098) and the Fundamental Research Funds for the Central Universities (300102321504, 300102321501, 300102321503).

  • Article
  • | |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • | |
  • Comments
    Abstract:

    Helmet wearing detection is susceptible to interference from complex background. To address this challenge, the YOLOv4 algorithm is employed. However, YOLOv4 faces issues such as slow detection speed, high memory consumption, computational complexity and demanding hardware performance requirements. This study integrates the MobileNet network to alleviate these challenges in YOLOv4. Additionally, cross-module feature fusion is introduced in MobileNet network, enabling effective fusion of high-level semantic features and low-level semantic features. Despite these advancements, small targets in images poses problems such as low resolution, limited informative features, coexistence of multiple scales, and potenial loss of feature information during continuous convolution. To mitigate these issues, this paper introduces neck optimization strategies, such as improving the feature pyramid FPN and introducing/ improving attention mechanism. These strategies focus on target information and reduce interference from background information during helmet detection. Simulation results show that the improved YOLOv4 neck-optimized network achieves a detection speed of 34.28FPS on the CPU platform, which is about 16 times faster than YOLOv4 network. Moreover, its detection accuracy is 4.21% higher than that of the YOLOv4 algorithm. This optimized algorithm strikes a balance between speed and accuracy.

    Reference
    Related
    Cited by
Get Citation

徐先峰,王轲,马志雄,姚景杰,赵万福.基于改进YOLOv4颈部优化网络的安全帽佩戴检测方法[J].重庆大学学报,2023,46(12):43~54

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:July 02,2022
  • Online: December 19,2023
Article QR Code