[发明专利]图像识别方法、装置、电子设备及计算机可读存储介质

说明书

本申请实施例提供了一种图像识别方法、装置、电子设备及计算机可读存储介质，涉及计算机技术领域。该方法包括：确定目标视频序列中当前处理的目标帧图像，将目标帧图像输入至第一识别模型，获得第一识别模型对目标帧图像进行滑窗切图处理后，输出的目标滑窗子图在目标帧图像中的第一位置信息，然后将目标帧图像以及第一位置信息输入至第二识别模型，获得第二识别模型基于第一位置信息对目标帧图像进行锚框切图处理后，输出的各目标对象在目标帧图像中的坐标位置。本申请实施例先基于滑窗切图确定可能存在目标对象的区域位置，再基于锚框切图对区域位置进行精准识别，能够提升图像识别的准确率，减少图像识别过程中的计算量。

技术领域

本申请涉及计算机技术领域，具体而言，本申请涉及一种图像识别方法、装置、电子设备及计算机可读存储介质。

背景技术

多目标跟踪(Multi Object Tracking，MOT)的主要任务是根据给定的图像序列，找到图像序列中的运动目标，并根据目标外观特征、位置、运动状态等信息，将不同图像帧中检测出的不同运动目标，与不同的identity(ID)进行一一关联，最终得到不同目标的运动轨迹。

大场景图像序列包含了更丰富的目标信息，可以对大规模时空范围内的目标及目标活动进行长时间、长距离的分析。但由于其图像覆盖范围广、光照等原因造成同一目标在不同时刻的外观存在较大差异；而且部分场景目标密集，尺度跨度大，存在严重遮挡、模糊现象，匹配难度大；视频帧率较高、像素极高，使得传统算法的识别准确率较低、计算量较大。

发明内容

本申请实施例的目的旨在能解决对高帧率图像中的目标轨迹识别准确率低、计算量大的问题。

根据本申请实施例的一个方面，提供了一种图像识别方法，该方法包括：

确定目标视频序列中当前处理的目标帧图像，目标帧图像中包括至少一个目标对象；

将目标帧图像输入至第一识别模型，获得第一识别模型对目标帧图像进行滑窗切图处理后，输出的目标滑窗子图在目标帧图像中的第一位置信息，目标滑窗子图为包括至少一个目标对象的滑窗子图；

将目标帧图像以及第一位置信息输入至第二识别模型，获得第二识别模型基于第一位置信息对目标帧图像进行锚框切图处理后，输出的各目标对象在目标帧图像中的坐标位置。

可选地，方法还包括：

根据各目标对象在目标帧图像中的坐标位置，确定目标帧图像中各目标对象的特征信息；

确定参考图像中已识别对象的特征信息以及移动轨迹信息；

根据各目标对象的特征信息、各目标对象在目标帧图像中的坐标位置、已识别对象的特征信息以及移动轨迹信息，确定各目标对象的移动轨迹信息；

其中，参考图像为目标视频序列中位于目标帧图像之前的帧图像；已识别对象的移动轨迹信息包括已识别对象在相应的参考图像中的坐标位置，以及基于已识别对象在参考图像中的坐标位置，预测得到的已识别对象在目标帧图像中的预测坐标位置。

可选地，根据各目标对象的特征信息、各目标对象在目标帧图像中的坐标位置、已识别对象的特征信息以及移动轨迹信息，确定各目标对象的移动轨迹信息，包括：

对于任意一个目标对象，若根据目标对象的特征信息以及各已识别对象的特征信息，确定存在与目标对象匹配的已识别对象，则根据目标对象在目标帧图像中的坐标位置，以及匹配的已识别对象的移动轨迹信息，获得目标对象的移动轨迹信息。

可选地，目标对象的移动轨迹信息包括目标对象在预测图像中的预测坐标位置；预测图像为目标视频序列中位于目标帧图像之后的帧图像。

可选地，根据各目标对象的特征信息、各目标对象在目标帧图像中的坐标位置、已识别对象的特征信息以及移动轨迹信息，确定各目标对象的移动轨迹信息，还包括：