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
Flowchart of YOLOv5s algorithm

Figure 2.

Multi-scale detection structure
Multi-scale detection structure

Figure 3.

Structure of CBAM module
Structure of CBAM module

Figure 4.

Channel Module
Channel Module

Figure 5.

Space module
Space module

Figure 6.

Schematic diagram of intersection and concatenation of IoU prediction and real frames.
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
GIoU penalty content for minimising the area of the shaded region

Figure 8.

Computational schematic of MPDIoU
Computational schematic of MPDIoU

Figure 9.

Comparison between before and after algorithm improvement
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
Published by: Xi’an Technological University
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.