[发明专利]人体部位分割网络训练方法、人体部位分割方法及装置

说明书

本发明实施例提供了一种人体部位分割网络训练方法、人体部位分割方法及装置，其中人体部位分割网络训练方法，包括获取训练人体图像数据及训练人体图像的训练人体特征，用待训练的人体部位分割网络获取训练人体部位分割标记，利用训练人体部位分割标记，获取基准评价分数，利用待训练的评价网络得到训练人体部位分割标记的训练评价分数，根据利用基准人体部位分割标记和所述训练人体部位分割标记获取的分割损失以及利用所述训练评价分数和所述基准评价分数获取的评价损失，获取模型训练的全局损失，进而优化人体部位分割网络和评价网络，得到训练完成的所述人体部位分割网络和所述评价网络，可以提高所述人体部位分割网络的准确性。

技术领域

本发明实施例涉及图像识别领域，尤其涉及一种人体部位分割网络训练方法、人体部位分割方法及装置。

背景技术

人体部位分割，属于语义分割任务的子任务，其目的主要是将包含有人体的图像进行细粒度分割，例如将人体图像分割为头部区域、躯干区域、四肢区域等各个部位的图像。

人体部位分割技术在诸多领域均有应用，如人体外观转移、行为识别、行人再识别、时装合成等等。因此，人体部位分割具有重要的研究意义和应用价值。

现有技术中的人体部位分割技术所得到的人体部位分割的准确性较差。

所以，如何提高人体部位分割的准确性，就成为本领域技术人员急需解决的技术问题。

发明内容

本发明实施例解决的技术问题是如何提高人体部位分割结果的质量。

为解决上述问题，本发明实施例提供了一种人体部位分割网络训练方法，包括：

获取训练人体图像数据，所述训练人体图像数据包括训练人体图像和所述训练人体图像的基准人体部位分割标记，所述基准人体部位分割标记包括所述训练人体图像的各个像素点的人体部位类别；

获取所述训练人体图像的训练人体特征；

利用待训练的人体部位分割网络，根据所述训练人体特征，获取训练人体部位分割标记；

至少利用所述训练人体部位分割标记，获取所述训练人体部位分割标记的基准评价分数，其中，所述基准评价分数用于评价所述训练人体部位分割标记的分割结果；

利用待训练的评价网络，根据所述训练人体特征，得到所述训练人体特征对应的训练评价分数，其中，所述评价网络适于评价所述人体部位分割网络所得的所述训练人体部位分割标记；

根据利用所述基准人体部位分割标记和所述训练人体部位分割标记获取的分割损失以及利用所述训练评价分数和所述基准评价分数获取的评价损失，获取模型训练的全局损失，并根据所述全局损失，优化待训练的所述人体部位分割网络和所述评价网络，直至所述全局损失满足全局损失阈值，得到训练完成的所述人体部位分割网络和所述评价网络。

为解决上述问题，本发明实施例还提供了一种人体部位分割方法，包括：

获取单人待分割图像；

获取所述单人待分割图像的人体特征；

利用所述人体部位分割网络训练方法训练得到的所述人体部位分割网络，根据所述人体特征，获取所述单人待分割图像对应的人体部位分割结果。

为解决上述问题，本发明实施例还提供了一种人体部位分割网络训练装置，包括：

训练人体图像数据获取单元，适于获取训练人体图像数据，所述训练人体图像数据包括训练人体图像和所述训练人体图像的基准人体部位分割标记，所述基准人体部位分割标记包括所述训练人体图像的各个像素点的人体部位类别；

训练人体特征获取单元，适于获取所述训练人体图像的训练人体特征；