A framework named CBAM-EfficientNetB0 was proposed in this paper, aimed at addressing the issue of low accuracy in distracted driver behavior recognition under low-parameter conditions. CBAM-EfficientNetB0 integrates CBAM attention mechanisms, including Channel Attention Module and Spatial Attention Module. This enables the network to focus more on important feature information, suppress irrelevant or redundant features, thereby enhancing feature distinctiveness and expressiveness, while also reducing complexity and improving model training efficiency. By adopting the cross-entropy loss function to address multi-class classification problems, and after experimenting with the optimization of the model's optimizer to SGD, optimized learning rates, and momentum parameters were obtained, which not only improved recognition accuracy but also enhanced model convergence. CBAM-EfficientNetB0 achieves a 97.6% accuracy with a basic parameter set of 5.33 million on the State Farm Distracted Driver Detection dataset. The results demonstrate that compared to other four frameworks in the same category, it achieves better accuracy and performs well under low-parameter conditions. In the future, this technology will be able to assist drivers in safe driving.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Recognition Algorithm for Distracted Driving Behavior Based on CBAM-EfficientNetB0

  • Xin Shi,
  • Fen Li,
  • Guangqiang Lu,
  • Yanjing Xie,
  • Fangyan Dong,
  • Kewei Chen

摘要

A framework named CBAM-EfficientNetB0 was proposed in this paper, aimed at addressing the issue of low accuracy in distracted driver behavior recognition under low-parameter conditions. CBAM-EfficientNetB0 integrates CBAM attention mechanisms, including Channel Attention Module and Spatial Attention Module. This enables the network to focus more on important feature information, suppress irrelevant or redundant features, thereby enhancing feature distinctiveness and expressiveness, while also reducing complexity and improving model training efficiency. By adopting the cross-entropy loss function to address multi-class classification problems, and after experimenting with the optimization of the model's optimizer to SGD, optimized learning rates, and momentum parameters were obtained, which not only improved recognition accuracy but also enhanced model convergence. CBAM-EfficientNetB0 achieves a 97.6% accuracy with a basic parameter set of 5.33 million on the State Farm Distracted Driver Detection dataset. The results demonstrate that compared to other four frameworks in the same category, it achieves better accuracy and performs well under low-parameter conditions. In the future, this technology will be able to assist drivers in safe driving.