[发明专利]图像识别方法、训练方法、装置、系统及存储介质

说明书

本公开提供了一种图像识别方法及装置，适用于人工智能技术领域和物联网技术领域。该图像识别方法包括：记录从作业场地入口进入作业场地的车辆的车辆标识，得到包括至少一个车辆标识的车辆标识集合；采集在作业平台周围工作的车辆的待处理图像信息，其中，作业平台位于作业场地中；根据待处理图像信息与车辆标识集合中的车辆标识的预测匹配结果，识别待处理图像信息中目标车辆的目标车辆标识。本公开还提供了一种图像识别模型的训练方法、装置、计算机系统、计算机可读存储介质以及计算机程序产品。

技术领域

本公开涉及计人工智能技术领域和物联网技术领域，更具体地，涉及一种图像识别方法、训练方法、装置、计算机系统、计算机可读存储介质以及计算机程序产品。

背景技术

通过摄像装置获取视频、图片等图像信息，利用图像识别技术对图像信息进行处理，以快速识别图像信息中的文本信息被广泛地应用于交通运输、仓储物流等多个应用场景。例如可以通过拍摄行驶在高速公路的车辆的图像信息，通过识别图像信息中车辆的车牌来确定违反交通规则的车辆。

在实现本公开构思的过程中，发明人发现相关技术中至少存在以下技术问题：由于图像信息存在图像模糊或图像缺失等缺陷，可能导致从图像信息中识别的文本信息出现错误。

发明内容

有鉴于此，本公开提供了一种图像识别方法、训练方法、装置、计算机系统、计算机可读存储介质以及计算机程序产品。

本公开的一个方面提供了一种图像识别方法，包括：

记录从作业场地入口进入作业场地的车辆的车辆标识，得到包括至少一个车辆标识的车辆标识集合；

采集在作业平台周围工作的车辆的待处理图像信息，其中，上述作业平台位于上述作业场地中；

根据上述待处理图像信息与上述车辆标识集合中的车辆标识的预测匹配结果，识别上述待处理图像信息中目标车辆的目标车辆标识。

根据本公开的实施例，根据上述待处理图像信息与上述车辆标识集合中的车辆标识的预测匹配结果，识别上述待处理图像信息中目标车辆的目标车辆标识包括：

对上述待处理图像信息进行图像特征提取，得到图像特征向量，其中，上述待处理图像信息中记录有目标信息；

对上述车辆标识集合中的车辆标识进行文本特征提取，得到文本特征向量；

基于注意力机制融合上述图像特征向量与上述文本特征向量，得到针对上述待处理图像信息的图像加权特征向量；

基于上述图像加权特征向量和上述文本特征向量，确定上述待处理图像信息的预测匹配结果；以及

在上述预测匹配结果表征上述待处理图像信息与上述车辆标识匹配的情况下，将上述待处理图像信息中的目标信息识别为上述目标车辆标识。

根据本公开的实施例，对上述待处理图像信息进行图像特征提取，得到图像特征向量包括：

将上述待处理图像信息输入至第一图像特征提取层，以便提取上述待处理图像信息在第一方向的图像特征，输出第一图像特征向量；以及

将上述待处理图像信息输入至第二图像特征提取层，以便提取上述待处理图像信息在第二方向的图像特征，输出第二图像特征向量；

其中，上述图像特征向量包括上述第一图像特征向量与上述第二图像特征向量，上述第一方向与上述第二方向相垂直。

根据本公开的实施例，上述基于注意力机制融合上述图像特征向量与上述文本特征向量，得到针对上述待处理图像信息的图像加权特征向量包括：

将上述第一图像特征向量与上述文本特征向量输入至第一注意力层，输出第一图像加权特征向量；以及