Skip to main content
Have a personal or library account? Click to login
Improved Pedestrian Vehicle Detection for Small Objects Based on Attention Mechanism Cover

Improved Pedestrian Vehicle Detection for Small Objects Based on Attention Mechanism

By: Yanpeng Hao and  Chaoyang Geng  
Open Access
|Sep 2024

Figures & Tables

Figure 1.

Flowchart of YOLOv5s algorithm

Figure 2.

Multi-scale detection structure

Figure 3.

Structure of CBAM module

Figure 4.

Channel Module

Figure 5.

Space module

Figure 6.

Schematic diagram of intersection and concatenation of IoU prediction and real frames.

Figure 7.

GIoU penalty content for minimising the area of the shaded region

Figure 8.

Computational schematic of MPDIoU

Figure 9.

Comparison between before and after algorithm improvement

Comparison of different algorithms

ModelVolumemAP@0.5%
YOLOv5s14.086.9
YOLOv8s22.486.5
YOLOv6s37.483.0
YOLOv4245.979.5
SSD100.371.0
YOLOv5193.789.7
ours15.790.9

Incorporation of multiple attention mechanisms

Attention MechanismmAP%P/%R/%
+SE85.391.895.0
+ECA86.692.394.1
+CCA86.492.094.6
+SA-Net87.892.594.7
+MS-CAM87.592.795.2
+CBAM88.593.295.8

Effect of internal parameters on the model

Batch SizeAverage accuracyaccuracyrecall rateconfidence level
Mean average precisionPrecision P/%Rcall(math.)
mAP percent R/%Confidence/%
1087.492.096.186.0
1388.393.496.084.0
1688.793.795.084.0
1889.494.395.982.0
2090.395.295.786.0

Ablation experiments

MethodsXCLPerson/%Car/%mAP%
YOLOv5s 79.294.386.9
X-YOLOv5s 82.394.488.3
C-YOLOv5s 81.895.288.5
L-YOLOv5s 82.595.589.0
XC-YOLOv5s 83.196.289.6
XL-YOLOv5s 83.496.690.0
CL-YOLOv5s 83.996.390.1
XCL-YOLOv5s84.797.090.9
Language: English
Page range: 80 - 89
Published on: Sep 30, 2024
In partnership with: Paradigm Publishing Services
Publication frequency: 4 issues per year

© 2024 Yanpeng Hao, Chaoyang Geng, published by Xi’an Technological University
This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 License.