[发明专利]车辆行驶方向的识别方法、装置、计算机设备和存储介质

说明书

本申请涉及一种车辆行驶方向的识别方法、装置、计算机设备和存储介质。所述方法包括：获取车辆行驶视频；根据所述车辆行驶视频对目标车辆进行跟踪得到车辆跟踪序列；从所述车辆跟踪序列中提取首部图像集与尾部图像集；确定所述首部图像集与所述尾部图像集中每帧目标车辆图像对应的目标车辆的车身部位；根据所述车身部位识别所述目标车辆的行驶方向。采用本方法能够提高识别效率与准确性。

技术领域

本申请涉及数据处理技术领域，特别是涉及一种车辆行驶方向的识别方法、装置、计算机设备和存储介质。

背景技术

基于车辆行驶视频识别车辆行驶方向逐渐被应用在多种应用场景下，比如在视频剪辑场景下，需要基于车辆行驶视频识别各目标车辆的车辆行驶方向，以根据所识别出的车辆行驶方向进行视频剪辑，还比如在违章逆行车辆的检测场景下，需要基于车辆行驶视频识别各目标车辆的车辆行驶方向，以根据所识别出的车辆行驶方向判断各目标车辆是否存在违章逆行问题，从而检测出违章逆行车辆。由此，如何基于车辆行驶视频识别车辆行驶方向是值得关注的问题。目前，通常采用人工观察车辆行驶视频的方式来识别判定车辆行驶方向。但是，该种识别方式下，受限于人工效率，导致识别效率较低，而因视觉疲劳等原因会降低识别准确性，尤其是在待识别车辆行驶方向的目标车辆较多的情况下，会显著降低车辆行驶方向的识别准确性与效率。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高识别效率与准确性的车辆行驶方向的识别方法、装置、计算机设备和存储介质。

一种车辆行驶方向的识别方法，所述方法包括：

获取车辆行驶视频；

根据所述车辆行驶视频对目标车辆进行跟踪得到车辆跟踪序列；

从所述车辆跟踪序列中提取首部图像集与尾部图像集；

确定所述首部图像集与所述尾部图像集中每帧目标车辆图像对应的目标车辆的车身部位；

根据所述车身部位识别所述目标车辆的行驶方向。

在其中一个实施例中，所述根据所述车辆行驶视频对目标车辆进行跟踪得到车辆跟踪序列，包括：

从所述车辆行驶视频中提取第一视频帧，对所述第一视频帧进行车辆检测确定待跟踪的目标车辆；

从所述车辆行驶视频中提取第二视频帧，根据所述第二视频帧对所述目标车辆进行跟踪得到车辆跟踪序列。

在其中一个实施例中，所述对所述第一视频帧进行车辆检测确定待跟踪的目标车辆，包括：

将所述第一视频帧输入已训练好的车辆检测模型，得到每个候选车辆对应的候选检测框；

从所述候选检测框中筛选符合跟踪条件的目标检测框；

根据所述目标检测框从所述候选车辆中确定待跟踪的目标车辆。

在其中一个实施例中，所述根据所述第二视频帧对所述目标车辆进行跟踪得到车辆跟踪序列，包括：

将各所述第二视频帧输入已训练好的车辆跟踪模型，得到所述目标车辆在各所述第二视频帧中对应的跟踪检测框；

根据各所述跟踪检测框从相应第二视频帧中提取所述目标车辆对应的目标车辆图像，得到车辆跟踪序列。

在其中一个实施例中，所述从所述车辆跟踪序列中提取首部图像集与尾部图像集，包括：

从所述车辆跟踪序列的首部提取第一数量的目标车辆图像得到首部图像集；

从所述车辆跟踪序列的尾部提取第二数量的目标车辆图像得到尾部图像集。

在其中一个实施例中，所述确定所述首部图像集与所述尾部图像集中每帧目标车辆图像对应的目标车辆的车身部位，包括：