[发明专利]基于YOLO目标检测技术的智能车辆运动路况信息的检测方法

摘要

基于YOLO目标检测技术的智能车辆运动路况信息的检测方法，它属于智能车运动路况信息检测领域。本发明解决了传统的采集智能车辆运动路况信息的方法的准确率较低、速度较慢的问题。本发明通过将摄像头固定于一定的高度，以一定的角度采集智能车辆所在的路况信息，用心形和星形固定于智能车辆上表示智能车辆，用箭头表示障碍物，用长方形和三角形表示终点，通过加载训练好的模型并对其输入摄像头采集到的信息，能够实时检测出智能车辆所在的路况信息，帧数能够达到每秒40帧，准确率到95％以上。因为运用了深度学习，所以该方法具有鲁棒性高，准确率高，速度快的优点。本发明可以应用于智能车运动路况信息检测领域用。

主权项

1.基于YOLO目标检测技术的智能车辆运动路况信息的检测方法，其特征在于，该方法包括以下步骤：步骤一、利用ImageNet数据库对YOLO目标检测模型做预训练，预训练完成后保留YOLO目标检测模型的卷积层和最大池化层，在保留的最后一个卷积层后依次添加一个平均池化层和一个全连接层用于分类任务；载入预训练好的参数值，并去掉添加的平均池化层和全连接层，在最后一个卷积层后再依次添加四个卷积层和两个全连接层，初始化全连接层的权重值以及偏置值，得到预训练好的YOLO目标检测模型；步骤二、利用摄像头采集智能车辆路况信息图像，每张图像均包括智能车辆、障碍物和终点三类目标；步骤三、采用添加随机椒盐噪声、添加高斯噪声和平移的方式对步骤二采集的图像进行图像数据增广，将增广后的图像数据分为训练集和测试集两部分；步骤四、将训练集样本输入到预训练好的YOLO目标检测模型中，采用计算机图形处理器GPU训练预训练好的YOLO目标检测模型，训练过程中输出损失函数的损失值；用TensorBoard实时可视化当前的损失值和正确率，训练停止后加载训练的模型，用测试集进行测试，测试完成后将训练结果存入计算机硬盘作为最终的训练模型；步骤五、利用摄像头采集包含智能车辆运动路况信息的图像，将采集到的图像输入到最终的训练模型中进行目标检测。