2019年度  第6期


标题:基于YOLO的电梯轿厢中狗识别算法
作者:华志超 章国宝
作者单位:东南大学自动化学院,江苏 南京210096
关键字:电梯环境,狗检测,卷积神经网络,YOLO,梯度直方图
摘要:提出了一种基于采用YOLO卷积神经网络的检测电梯轿厢中是否有狗的算法。该算法结合YOLO卷积神经网络和梯度直方图HOG特征,首先统一数据集尺寸并预处理作为网络输入,然后通过YOLO网络进行训练,提取图像特征并融合狗的HOG特征,最后通过非极大值抑制定义损失函数对模型进行优化,实现特征输出。YOLO算法具有快速检测以及较好的泛化能力的特点。该算法最终实现了对电梯图像中狗的检测,且有较好的实时性和准确率。