基于YOLOv5的电梯行人遮挡检测

Elevator Pedestrian Occlusion Detection Based on YOLOv5

摘要: 本文基于YOLOv5对手扶电梯行人的遮挡进行检测，针对电梯场景中行人遮挡导致的检测难的问题，本文在YOLOv5算法的基础上进行了改进。首先，为了解决因遮挡造成的预测框高度重合问题，本文采用了软非极大值抑制(Soft NMS)算法替代传统的NMS算法，该算法通过先对预测框的得分进行衰减处理，然后再进行过滤，从而减少因遮挡导致的人物漏检的情况。其次，针对遮挡导致行人身体可见区域减小、易被忽略的问题，本文在检测端增加了一个微小目标检测头，专门用于检测因遮挡而只部分可见的行人目标，从而提高了模型对这类目标的检测能力。实验结果表明，通过采用以上两种的改进方法对YOLOv5算法进行改进，在电梯行人遮挡检测中能够取得较好的效果。

Abstract: This article is based on YOLOv5 to detect pedestrian occlusion in escalators. To address the problem of difficulty in detecting pedestrian occlusion in elevator scenes, this article improves the YOLOv5 algorithm. Firstly, in order to solve the problem of high overlap of predicted boxes caused by occlusion, this paper adopts the Soft Non Maximum Suppression (Soft NMS) algorithm instead of the traditional NMS algorithm. This algorithm attenuates the scores of predicted boxes first, and then filters them to reduce the occurrence of missing persons caused by occlusion. Secondly, in response to the problem of occlusion causing a reduction in the visible area of pedestrians and being easily overlooked, this paper adds a small object detection head at the detection end, specifically designed to detect pedestrian targets that are only partially visible due to occlusion, thereby improving the model's detection ability for such targets. The experimental results show that by using the above two improvement methods to improve the YOLOv5 algorithm, good results can be achieved in elevator pedestrian occlusion detection.