基于YOLO的电梯轿厢中狗识别算法

摘要

提出了一种基于采用YOLO卷积神经网络的检测电梯轿厢中是否有狗的算法.该算法结合YOLO卷积神经网络和梯度直方图HOG特征,首先统一数据集尺寸并预处理作为网络输入,然后通过YOLO网络进行训练,提取图像特征并融合狗的HOG特征,最后通过非极大值抑制定义损失函数对模型进行优化,实现特征输出.YOLO算法具有快速检测以及较好的泛化能力的特点.该算法最终实现了对电梯图像中狗的检测,且有较好的实时性和准确率.