| Literature DB >> 35885207 |
Lili Wang1, Wenjie Yao1, Chen Chen1, Hailu Yang1.
Abstract
In actual driving scenes, recognizing and preventing drivers' non-standard driving behavior is helpful in reducing traffic accidents. To resolve the problems of various driving behaviors, a large range of action, and the low recognition accuracy of traditional detection methods, in this paper, a driving behavior recognition algorithm was proposed that combines an attention mechanism and lightweight network. The attention module was integrated into the YOLOV4 model after improving the feature extraction network, and the structure of the attention module was also improved. According to the 20,000 images of the Kaggle dataset, 10 typical driving behaviors were analyzed, processed, and recognized. The comparison and ablation experimental results showed that the fusion of an improved attention mechanism and lightweight network model had good performance in accuracy, model size, and FLOPs.Entities:
Keywords: YOLOV4 model; attention mechanism; driving behavior recognition; feature extraction
Year: 2022 PMID: 35885207 PMCID: PMC9321050 DOI: 10.3390/e24070984
Source DB: PubMed Journal: Entropy (Basel) ISSN: 1099-4300 Impact factor: 2.738
Figure 1The depth-wise separable convolution.
Figure 2The linear bottleneck and inverted residual structure.
Figure 3The process of attention realization.
Figure 4The detection network of the driving behaviors.
Figure 5The improved channel attention structure.
Figure 6The improved spatial attention structure.
Figure 7Distracted driving behaviors.
Results of the ablation experiment.
| Model | mAP/% | Params/M | FLOPs/G |
|---|---|---|---|
| YOLOV4 | 7.93 | 64.363 | 60.527 |
| YOLOV4 + MobileNetV3 | 80.5 | 40.692 | 39.652 |
| YOLOV4 + SA + CA | 80.4 | 64.363 | 60.527 |
| Our algorithm | 96.49 | 12.629 | 10.652 |
Figure 8Heat map. (a) Original image; (b) Heap map byYOLOV4 + MobileNetV3; (c) Heap map byour algorithm.
Figure 9The scatter plot of the mAP changes during training.
The t-test.
| Significant Level | YOLOV4 & Our Algorithm | YOLOV4 + MobileNetV3 & Our Algorithm | YOLOV4 + SA + CA & Our Algorithm |
|---|---|---|---|
| α | 2 × 10−15 | 9.75 × 10−6 | 8.63 × 10−6 |
The results of the different algorithms.
| Model | mAP/% | Params/M | FLOPs/G |
|---|---|---|---|
| Drive-Net [ | 95 | - | - |
| ST-SRU [ | 95.6 | 2.863 | 7.42 |
| Tutor–Student [ | 96.29 | 34.71 | 11.4 |
| SSD [ | 94.65 | 26.285 | 119.131 |
| LSTM [ | 88.15 |
|
|
| Our algorithm |
| 12.629 | 10.652 |