Table 3 Experimental results of adding attention mechanism.
From: Research on helmet wearing detection method based on deep learning
Attention mechanism | mAP (IoU = 0.5) | mAP (IoU = 0.5:0.95) | P | R | FPS |
|---|---|---|---|---|---|
YOLOv5s | 0.819 | 0.573 | 0.87 | 0.751 | 96.2 |
CA | 0.812 | 0.558 | 0.884 | 0.74 | 88.5 |
CBAM | 0.821 | 0.575 | 0.879 | 0.758 | 84 |
SE | 0.828 | 0.566 | 0.87 | 0.759 | 94.3 |
