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

说明书

本公开涉及一种图像处理方法、装置、存储介质及电子设备。该方法包括：获取包含目标人体的待处理图像；将待处理图像输入目标人体图像分割模型中，对目标人体进行部位分割，以得到部位分割后的目标图像；其中，该目标人体图像分割模型为根据目标样本数据集训练得到的，目标样本数据集包括多个目标样本图像，以及每个目标样本图像对应的目标部位标签，目标样本图像包括第一待定样本数据集中的第一样本图像，目标部位标签为根据第一待定样本数据集和第二待定样本数据集获取的；第一待定样本数据集包括第一样本图像及其对应的第一人体关键点标签；第二待定样本数据集包括第二样本图像及其对应的第二人体关键点标签和第二部位标签。

技术领域

本公开涉及人工智能领域，具体地，涉及一种图像处理方法、装置、存储介质及电子设备。

背景技术

人体部位分割是在包含人体的图像中检索出更细粒度地语义标签，例如头部标签、四肢标签和躯干标签等。人体部位分割的应用范围广泛，例如，可以应用在行为理解、机器人操作、启示推理和识别人机交互等场景。

在相关技术中，可以通过神经网络模型实现人体部位分割，但是为了对神经网络模型进行训练，需要大量经过标注的训练样本数据，而对样本数据进行标注的成本极高，导致难以获取足够的样本数据对神经网络模型进行训练，样本量少会导致训练后的模型难以准确实现人体部位分割。

发明内容

本公开的目的是提供一种图像处理方法、装置、存储介质及电子设备，以部分地解决相关技术中存在的上述问题。

为了实现上述目的，本公开的第一方面提供了一种图像处理方法，所述方法包括：

获取包含目标人体的待处理图像；

将所述待处理图像输入目标人体图像分割模型中，对所述目标人体进行部位分割，以得到部位分割后的目标图像；

其中，所述目标人体图像分割模型为根据目标样本数据集训练得到的，所述目标样本数据集包括多个目标样本图像，以及每个所述目标样本图像对应的目标部位标签，所述目标样本图像包括第一待定样本数据集中的第一样本图像，所述目标部位标签为根据所述第一待定样本数据集和第二待定样本数据集获取的；所述第一待定样本数据集包括多个第一样本图像，以及每个所述第一样本图像对应的第一人体关键点标签；所述第二待定样本数据集包括多个第二样本图像，以及每个所述第二样本图像对应的第二人体关键点标签和第二部位标签。

可选地，所述目标人体图像分割模型是通过以下方式训练得到的：

获取所述第一待定样本数据集和所述第二待定样本数据集；

根据所述第二待定样本数据集获取所述第一样本图像对应的第一部位标签；

根据所述第一样本图像和所述第一部位标签，确定所述目标样本数据集；

根据所述目标样本数据集，对第一预设神经网络模型进行训练，得到所述目标人体图像分割模型。

可选地，所述根据所述第二待定样本数据集获取所述第一样本图像对应的第一部位标签包括：

根据所述第一人体关键点标签，获取所述第一待定样本数据集中每个所述第一样本图像对应的第一人体姿态；

根据所述第二人体关键点标签，获取所述第二待定样本数据集中每个所述第二样本图像对应的第二人体姿态；

根据所述第一人体姿态和所述第二人体姿态，确定所述第一样本图像对应的第一部位标签。

可选地，所述根据所述第一人体姿态和所述第二人体姿态，确定所述第一样本图像对应的第一部位标签包括：

根据标签获取步骤，获取所述第一样本图像对应的第一候选部位标签；

根据所述第一候选部位标签，确定所述第一样本图像对应的第一部位标签；