Table 3 Comparison of ablation experiment results.
From: Detection of safety helmet and mask wearing using improved YOLOv5s
Algorithm model | ShuffleNetv2 | ECA | EIoU | mAP@0.5 | Parameter volume/M | Computing volume/M | Speed/s |
|---|---|---|---|---|---|---|---|
YOLOv5s | – | – | – | 0.832 | 7.05 | 14.5 | 0.0086 |
YOLOv5s-1 | √ | – | – | 0.805 | 3.12 | 6.2 | 0.0074 |
YOLOv5s-2 | √ | √ | – | 0.825 | 3.16 | 6.4 | 0.0082 |
YOLOv5s-3 | √ | √ | √ | 0.875 | 3.19 | 6.7 | 0.0096 |