一种基于YOLO的车辆信息检测和跟踪方法、装置及计算机存储介质制造方法及图纸

本申请实施例提供了一种基于YOLO的车辆信息检测和跟踪方法、装置及计算机存储介质，其中，该方法包括：获取待检测的视频数据；通过预设的基于YOLO的车辆目标检测模型对待测图像进行检测，以确定待测图像中的目标车辆；通过预设的基于YOLO的车辆车标检测模型、预设的基于YOLO的车辆颜色检测模型对待测图像进行检测，以获取目标车辆车标检测结果、颜色检测结果；根据目标车辆车标检测结果和/或目标车辆颜色检测结果锁定目标车辆，并通过数据关联对目标车辆进行跟踪。由于该方案通过利用基于YOLO的检测模型直接对待测图像进行检测，识别出目标车辆，以及目标车辆的车标和颜色，进而准确地锁定目标车辆进行相应跟踪，从而解决了现有方法存在的实施过程复杂、处理效率低的技术问题。

A Method, Device and Computer Storage Media for Vehicle Information Detection and Tracking Based on YOLO

The embodiment of this application provides a vehicle information detection and tracking method, device and computer storage medium based on YOLO, in which the method includes: acquiring video data to be detected; detecting the detected image through a preset YOLO-based vehicle target detection model to determine the target vehicle in the image to be detected; Through the preset YOLO-based vehicle logo detection model and the preset YOLO-based vehicle color detection model, the image to be detected is detected to obtain the target vehicle logo detection results and color detection results; according to the target vehicle logo detection results and/or the target vehicle color detection results, the target vehicle is locked and passed. Data association is used to track the target vehicle. Because the scheme uses YOLO-based detection model to detect the image directly, identifies the target vehicle, as well as the vehicle logo and color of the target vehicle, and then locks the target vehicle accurately for corresponding tracking, thus solving the technical problems of complex implementation process and low processing efficiency existing methods.

【技术实现步骤摘要】
一种基于YOLO的车辆信息检测和跟踪方法、装置及计算机存储介质
本申请涉及计算机视觉
，特别涉及一种基于YOLO的车辆信息检测和跟踪方法、装置及计算机存储介质。
技术介绍
随着技术的发展与普及，利用计算机视觉技术自动检测识别范围区内的车辆在城市监控、交通疏导、智能驾驶等领域逐渐受到越来越多的关注和重视。目前，现有的车辆跟踪方法大多是通过基于相应的特征提取方案（例如HOG、SURF等）对先将图像分成多个图像区块，再分别对多个图像区块中的车辆特征（例如车牌等特征）分别进行识别检测，再综合从多个图像区块中识别出的车辆特征，确定目标车辆的基本特征信息，例如确定出目标车辆的车牌信息，再结合目标车辆的基本信息锁定目标车辆，对目标车辆进行进一步的跟踪。上述方法由于受限于实现机理，往往不能直接对待测图像进行识别检测，而必须先分区块再综合识别，因此实施过程较为繁琐。此外，现有方法实施时大多需要依赖某些具体的识别算法，而这种识别算法通常较为复杂，导致处理时处理效率相对较低。综上可知，现有方法具体实施时，往往存在的实施过程复杂、处理效率低、占用资源量大的技术问题。针对上述问题，目前尚未提出有效的解决本文档来自技高网...

【技术保护点】
1.一种基于YOLO的车辆信息检测和跟踪方法，其特征在于，包括：获取待检测的视频数据，其中，所述待检测的视频数据包括多帧待测图像；通过预设的基于YOLO的车辆目标检测模型对所述待测图像进行检测，以确定待测图像中的目标车辆；通过预设的基于YOLO的车辆车标检测模型、预设的基于YOLO的车辆颜色检测模型对所述待测图像进行检测，以获取目标车辆车标检测结果、目标车辆颜色检测结果；根据所述目标车辆车标检测结果和/或所述目标车辆颜色检测结果锁定所述目标车辆，并通过数据关联对待检测的视频数据中的目标车辆进行跟踪。

【技术特征摘要】
1.一种基于YOLO的车辆信息检测和跟踪方法，其特征在于，包括：获取待检测的视频数据，其中，所述待检测的视频数据包括多帧待测图像；通过预设的基于YOLO的车辆目标检测模型对所述待测图像进行检测，以确定待测图像中的目标车辆；通过预设的基于YOLO的车辆车标检测模型、预设的基于YOLO的车辆颜色检测模型对所述待测图像进行检测，以获取目标车辆车标检测结果、目标车辆颜色检测结果；根据所述目标车辆车标检测结果和/或所述目标车辆颜色检测结果锁定所述目标车辆，并通过数据关联对待检测的视频数据中的目标车辆进行跟踪。2.根据权利要求1所述的方法，其特征在于，所述基于YOLO的车辆目标检测模型按照以下方式建立：获取第一样本数据，其中，所述第一样本数据为包含有车辆的图像数据；对所述第一样本数据进行车辆标注，并将标注后的第一样本数据分为第一训练集和第一测试集；利用所述第一训练集对YOLO神经网络进行训练，并利用所述第一测试集对所述训练后的YOLO神经网络进行测试，以建立所述基于YOLO的车辆目标检测模型。3.根据权利要求2所述的方法，其特征在于，所述第一样本数据具体可以为包含有真实车辆形状和大小的图像数据，或包含有虚拟车辆形状和大小的图像数据，或包含有真实车辆形状和大小的图像数据和包含有虚拟车辆形状和大小的图像数据的合集。4.根据权利要求1所述的方法，其特征在于，所述基于YOLO的车辆车标检测模型按照以下方式建立：获取第二样本数据，其中，所述第二样本数据为包含有车辆车标的车辆图像数据；对所述第二样本数据进行车标标注，并将标注后的第二样本数据分为第二训练集和第二测试集；利用所述第二训练集对YOLO神经网络进行训练，并利用所述第二测试集对所述训练后的YOLO神经网络进行测试，以建立所述基于YOLO的车辆车标检测模型；对于第二样本数据，其特征在于，所述包含有车辆车标的车辆图像数据为车辆的车头或车尾照片。5.根据权利要求1所述的方法，其特征在于，所述基于YOLO的车辆颜色检测模型按照以下方式建立：获取第三样本数据，其中，所述第三样本数据为包含有车辆颜色的车辆图像数据；对所述第三样本数据进行颜色标注，并将标注后的第三样本数据分为第三训练集和第三测试集；利用所述第三训练集对YOLO神经网络进行训练，...

【专利技术属性】
技术研发人员：王福建，李建元，温晓岳，陈涛，沈坚，
申请(专利权)人：银江股份有限公司，
类型：发明
国别省市：浙江,33