[发明专利]车辆匹配样本生成方法、装置、计算机设备和存储介质

说明书

本申请涉及一种车辆匹配样本生成方法、装置、计算机设备和存储介质。该方法包括：获取至少一个第一锚框和标注框；根据第一锚框与车身标注框之间的交并比，从至少一个第一锚框中确定出车身样本框；其中，车身样本框包括车身正样本框；对车身正样本框进行分割，得到预设数量的第二锚框；根据第二锚框与车头标注框、车尾标注框之间的交并比，从预设数量个第二锚框中确定出车头样本框和车尾样本框；其中，车身样本框、车头样本框和车尾样本框之间属于同一车辆。从而能够快速地从图像中确定出同一车辆的车身、车头、车尾，可以提高样本标注效率和标注的准确率。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种车辆匹配样本生成方法、装置、计算机设备和存储介质。

背景技术

随着计算机视觉技术的发展，出现了识别图像中同一车辆的车身、车头、车尾和车身其他部位的技术，应用该技术可以实现对车辆的定位及跟踪。传统方法中，通常是通过训练一个能够识别车身、车头和车尾的模型，通过该模型识别车身、车头和车尾，并通过预设的匹配规则确定出车身、车头和车尾之间的关联关系。

然而，传统方法中，在训练模型之前，需要人工从图像中标注出车身、车头、车尾及各部分之间关联关系，人工标注的方式需要耗费大量的人力和时间，且标注准确率低。

发明内容

基于此，有必要针对上述技术问题，提供一种能够提高样本标注准确率的车辆匹配样本生成方法、装置、计算机设备和存储介质。

一种车辆匹配样本生成方法，所述方法包括：

获取至少一个第一锚框和标注框；其中，所述标注框为对图像中的每一车辆的车辆类型进行标注生成的框，所述车辆类型包括车身、车头和车尾，所述标注框包括车身标注框、车头标注框和车尾标注框；

根据所述第一锚框与所述车身标注框之间的交并比，从所述至少一个第一锚框中确定出车身样本框；其中，所述车身样本框包括车身正样本框和车身负样本框；

对所述车身正样本框进行分割，得到预设数量的第二锚框；

根据所述第二锚框与所述车头标注框、所述车尾标注框之间的交并比，从所述预设数量个第二锚框中确定出车头样本框和车尾样本框；其中，所述车身样本框、所述车头样本框和所述车尾样本框之间属于同一车辆。

在其中一个实施例中，所述根据所述第一锚框与所述车身标注框之间的交并比，从所述至少一个第一锚框中确定出车身样本框，包括：

对于每一所述第一锚框，获取所述第一锚框与所述车身标注框之间的交并比，得到第一交并比；

检测所述第一交并比与第一预设阈值之间的大小关系；

若所述第一交并比大于所述第一预设阈值，则将所述第一锚框确定为所述车身正样本框，否则将所述第一锚框确定为所述车身负样本框。

在其中一个实施例中，所述根据所述第二锚框与所述车头标注框、所述车尾标注框之间的交并比，从所述预设数量个第二锚框中确定出车头样本框和车尾样本框，包括：

获取所述第二锚框与所述车头标注框之间的交并比，得到第二交并比；

获取所述第二锚框与所述车尾标注框之间的交并比，得到第三交并比；

根据所述第二交并比与所述第三交并比之间的大小关系，从所述预设数量个第二锚框中确定出车头样本框和车尾样本框。

在其中一个实施例中，所述根据所述第二交并比与所述第三交并比之间的大小关系，从所述预设数量个第二锚框中确定出车头样本框和车尾样本框，包括：

检测所述第二交并比与所述第三交并比之间的大小关系；

若所述第二交并比大于所述第三交并比，则将所述第二锚框确定为所述车头样本框，否则将所述第二锚框确定为所述车尾样本框。