[发明专利]车辆检测及标注方法、系统及电子设备

说明书

本发明提供一种车辆检测及标注方法、系统及电子设备，所述车辆检测及标注系统包括：视频流追踪系统和视频流标注系统；所述视频流追踪系统用于追踪一段视频流中出现的车辆轨迹，逐帧得到视频流中车辆的位置坐标；所述视频流标注系统用于对追踪出的车辆进行车型及颜色的标记，并标记视频中同一辆车出现的所有帧。本发明可以充分利用车辆识别和追踪技术的优势，在保证可用的标注准确率的同时，免去逐帧标记视频流中车辆的繁琐和效率低下。

技术领域

本发明涉及车辆识别、车辆追踪及数据标注领域，特别设计一种车辆检测及标注方法、 系统及电子设备。

背景技术

随着近年来图像识别和深度学习技术的不断发展，许多领域需要训练自己的汽车识别模 型。但由于各种领域对汽车识别模型的需求不同，很多情况下需要对已有的模型进行数据集 增强，或者重新用车辆数据训练自己的模型。

传统的车辆数据标注采用的是用图像或者视频逐帧进行手工标注。在例如LabelImg、 OpenCV/CVAT、Ybat等工具的标注过程中，标注者需要逐帧在图像上勾画车辆的包围盒或 轮廓。这样的标注优势是准确率高，但是也存在一个十分明显的缺点，那就是效率较为低下。

如果使用车辆追踪技术对追踪出的车辆进行标注，在建立轨迹联系后只要对一个识别框 进行标注，后续所有的连续的识别框都能被标注上。理想情况下一个视频中出现了几辆车， 就只需要进行几次标注。一辆车常常在一个视频中出现数百帧甚至更多，使用车辆追踪技术 无疑能大幅提高标注效率。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种车辆检测及标注方法、系统 及电子设备，用于解决现有技术无法车辆数据标注效率低、费时费力的技术问题。

为实现上述目的及其他相关目的，本发明提供一种车辆检测及标注方法，包括：追踪一 段视频流中出现的车辆轨迹，逐帧得到视频流中车辆的位置坐标；在视频流被追踪完毕后， 创建一个可被视频流标注系统读取的项目，并为后续的标注提供车辆的追踪信息；对追踪出 的车辆进行车型及颜色的标记，并标记视频中同一辆车出现的所有帧。

于本发明的一实施例中，所述追踪一段视频流中出现的车辆轨迹，逐帧得到视频流中车 辆的位置坐标包括：根据车辆识别模型输出的位置坐标，以及视频流中的车辆特征信息进行 匹配及预测；逐帧识别视频流中车辆的位置坐标。

于本发明的一实施例中，所述标记视频中同一辆车出现的所有帧包括：标注追踪出的一 条车辆轨迹；手工标注未追踪出的车辆轨迹；删除错误的车辆追踪包围盒、切割轨迹连续的 数辆车。

于本发明的一实施例中，所述车辆检测及标注方法还包括：将视频剪切成原视频的一部 分；根据识别出的帧序号和车辆位置坐标，对视频进行渲染，并使用亮色标记出视频中的车 辆信息。

本发明的实施例还提供一种车辆检测及标注系统，包括：视频流追踪系统和视频流标注 系统；所述视频流追踪系统用于追踪一段视频流中出现的车辆轨迹，逐帧得到视频流中车辆 的位置坐标；所述视频流标注系统用于对追踪出的车辆进行车型及颜色的标记，并标记视频 中同一辆车出现的所有帧。

于本发明的一实施例中，还包括分别与所述视频流追踪系统和视频流标注系统连接的车 辆数据保存系统；所述车辆数据保存系统用于在视频流被追踪完毕后，创建一个可被所述视 频流标注系统读取的项目，并为后续的标注提供车辆的追踪信息。

于本发明的一实施例中，所述视频流追踪系统包括车辆识别模型和车辆追踪模型；所述 车辆识别模型用于逐帧识别视频流中车辆的位置坐标；所述车辆追踪模型用于根据车辆识别 模型输出的位置坐标，以及视频流中的车辆特征信息进行匹配及预测。