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

权利要求书

1.一种车辆行驶方向的识别方法，其特征在于，所述方法包括：

获取车辆行驶视频；

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

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述根据所述车辆行驶视频对目标车辆进行跟踪得到车辆跟踪序列，包括：

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

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

3.根据权利要求2所述的方法，其特征在于，所述对所述第一视频帧进行车辆检测确定待跟踪的目标车辆，包括：

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

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

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

4.根据权利要求2所述的方法，其特征在于，所述根据所述第二视频帧对所述目标车辆进行跟踪得到车辆跟踪序列，包括：

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

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

5.根据权利要求1至4任意一项所述的方法，其特征在于，所述从所述车辆跟踪序列中提取首部图像集与尾部图像集，包括：

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

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

6.根据权利要求1至4任意一项所述的方法，其特征在于，所述确定所述首部图像集与所述尾部图像集中每帧目标车辆图像对应的目标车辆的车身部位，包括：

将所述首部图像集与所述尾部图像集中的每帧目标车辆图像，输入已训练好的分类模型，得到所述目标车辆在每帧目标车辆图像对应的车身部位。

7.根据权利要求1至4任意一项所述的方法，其特征在于，所述根据所述车身部位识别所述目标车辆的行驶方向，包括：

根据所述目标车辆在所述首部图像集的每帧目标车辆图像中对应的车身部位，对各车身部位进行投票，将票数最多的车身部位确定为所述目标车辆在所述首部图像集中对应的车身部位；

根据所述目标车辆在所述尾部图像集的每帧目标车辆图像中对应的车身部位，对各车身部位进行投票，将票数最多的车身部位确定为所述目标车辆在所述尾部图像集中对应的车身部位；

根据所述目标车辆在所述首部图像集与所述尾部图像集中所对应的车身部位，识别所述目标车辆的行驶方向。

8.一种车辆行驶方向的识别装置，其特征在于，所述装置包括：

获取模块，用于获取车辆行驶视频；

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

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

分类模块，用于确定所述首部图像集与所述尾部图像集中每帧目标车辆图像对应的目标车辆的车身部位；

识别模块，用于根据所述车身部位识别所述目标车辆的行驶方向。

9.一种计算机设备，包括存储器和处理器，所述存储器存储有计算机程序，其特征在于，所述处理器执行所述计算机程序时实现权利要求1至7中任一项所述的方法的步骤。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1至7中任一项所述的方法的步骤。