HRM: An Intelligent Helmet Recognition Model in Complex Scenes

Figure 3

(a) Adding SE-Net in the YOLOv5 backbone layer. (b) Adding SE-Net in the YOLOv5 neck part. SE (128) means that this layer’s input size is 128.

(a)

(b)