[发明专利]图像处理方法、装置、电子设备和存储介质

说明书

本公开实施例公开了一种图像处理方法、装置、电子设备和存储介质。所述方法包括：获得多帧图像；对所述多帧图像中的第一图像中的目标对象进行肢体关键点检测处理，获得所述目标对象的部分肢体对应的第一关键点信息；基于所述第一关键点信息确定第二图像中的所述目标对象的所述部分肢体对应的第二关键点信息；其中，所述第一图像为所述多帧图像中的任意一帧图像；所述第二图像为所述第一图像后的一帧图像。

技术领域

本公开涉及计算机视觉技术领域，具体涉及一种图像处理方法、装置、电子设备和存储介质。

背景技术

目标跟踪技术通常基于肢体检测算法和肢体关键点检测算法，利用肢体检测算法检测出的人体，以及肢体关键点检测算法检测出的人体关键点，实现目标跟踪。但是目前的肢体检测算法和肢体关键点检测算法无法适应只有上半身肢体的场景，从而导致只有上半身肢体的目标无法进行跟踪。

发明内容

本公开实施例提供一种图像处理方法、装置、电子设备和存储介质。

本公开实施例提供了一种图像处理方法，所述方法包括：获得多帧图像；对所述多帧图像中的第一图像中的目标对象进行肢体关键点检测处理，获得所述目标对象的部分肢体对应的第一关键点信息；基于所述第一关键点信息确定第二图像中的所述目标对象的所述部分肢体对应的第二关键点信息；其中，所述第一图像为所述多帧图像中的任意一帧图像；所述第二图像为所述第一图像后的一帧图像。

上述方案中，所述对第一图像中的目标对象进行肢体关键点检测处理，获得所述目标对象的部分肢体对应的第一关键点信息，包括：对所述第一图像中的所述目标对象进行肢体检测处理，确定所述目标对象的第一区域；所述第一区域包括所述目标对象的部分肢体所在区域；对所述第一区域对应的像素点进行肢体关键点检测处理，获得所述目标对象的所述部分肢体对应的第一关键点信息。

上述方案中，所述基于所述第一关键点信息确定第二图像中的所述目标对象的所述部分肢体对应的第二关键点信息，包括：基于所述第一关键点信息在所述第一图像中确定第二区域；所述第二区域大于所述目标对象的第一区域；所述第一区域包括所述目标对象的部分肢体所在区域；根据所述第二区域或所述第一关键点信息，确定所述第二图像中与所述第二区域的位置范围对应的第三区域；对所述第二图像中的所述第三区域内的像素点进行肢体关键点检测处理，获得所述部分肢体对应的第二关键点信息。

上述方案中，所述对所述第一图像中的所述目标对象进行肢体检测处理，包括：利用肢体检测网络对所述第一图像中的所述目标对象进行肢体检测处理；其中，所述肢体检测网络采用第一类样本图像训练得到；所述第一类样本图像中标注有目标对象的检测框；所述检测框的标注范围包括所述目标对象的部分肢体所在区域。

上述方案中，所述对所述第一区域对应的像素点进行肢体关键点检测处理，包括：利用肢体关键点检测网络对所述第一区域对应的像素点进行肢体关键点检测处理；其中，所述肢体关键点检测网络采用第二类样本图像训练得到；所述第二类样本图像中标注有包括所述目标对象的部分肢体的关键点。

上述方案中，所述目标对象的部分肢体包括以下至少之一：头部、颈部、肩部、胸部、腰部、髋部、手臂、手部；所述第一关键点信息和所述第二关键点信息包括头部、颈部、肩部、胸部、腰部、髋部、手臂和手部中的至少一个肢体的轮廓关键点信息和/骨骼关键点信息。

上述方案中，所述方法还包括：响应于获得所述目标对象的部分肢体对应的第一关键点信息的情况，为所述目标对象分配跟踪标识；基于对所述多帧图像的处理过程中分配的所述跟踪标识的数量，确定所述多帧图像中的目标对象的数量。

上述方案中，所述方法还包括：基于所述第二关键点信息确定所述目标对象的姿态；基于所述目标对象的姿态确定对应于所述目标对象的交互指令。