[发明专利]图像处理模型的训练方法、服装效果图的生成方法及装置

说明书

本发明实施例公开了一种图像处理模型的训练方法、服装效果图的生成方法及装置，该方法包括：将获取到的待训练服装图像集中的各待训练服装图像分别输入到初始图像处理模型中；通过特征提取网络，对待训练服装图像进行特征提取，得到图像特征；通过关键点定位网络，基于特征提取网络输出的图像特征，输出预测服装关键点；通过图像分割网络，基于特征提取网络输出的图像特征，输出预测服装分割图像；基于各待训练服装图像分别对应的预测服装关键点、预测服装分割图像以及标准训练数据，对初始图像处理模型的模型参数进行调整，直到得到训练完成的目标图像处理模型。本发明实施例解决了现有服装图像的处理过程中步骤繁多的问题。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种图像处理模型的训练方法、服装效果图的生成方法及装置。

背景技术

随着电子商务的发展，线上购买服装已经成为一种潮流，虚拟试衣技术是线上服装销售的重要研究内容。

在虚拟试衣技术中，通常需要对用户提供的待检测服装图像执行图像分割操作，得到仅包含服装的服装分割图像，以及对待检测服装图像执行关键点定位操作，得到服装图像中的服装关键点，基于服装分割图像和服装关键点，生成服装穿在模特身上的服装效果图。

在实现本发明的过程中，发现现有技术中至少存在以下技术问题：

在对待检测服装图像进行处理的过程中，现有的图像处理算法仅能完成一种图像处理任务，从而导致整个图像处理过程的步骤繁多，且每个图像处理算法的处理结果的精确度都不高。

发明内容

本发明实施例提供了一种图像处理模型的训练方法、服装效果图的生成方法及装置，以解决现有服装图像的处理过程中步骤繁多的问题，提高图像处理算法的处理结果的精确度。

根据本发明一个实施例提供的一种图像处理模型的训练方法，该方法包括：

将获取到的待训练服装图像集中的各待训练服装图像分别输入到初始图像处理模型中；其中，所述初始图像处理模型包括特征提取网络、关键点定位网络和图像分割网络；

针对每个待训练服装图像，通过所述特征提取网络，对所述待训练服装图像进行特征提取，得到图像特征；

通过所述关键点定位网络，基于所述特征提取网络输出的图像特征，输出所述待训练服装图像对应的预测服装关键点；

通过所述图像分割网络，基于所述特征提取网络输出的图像特征，输出所述待训练服装图像对应的预测服装分割图像；

基于各所述待训练服装图像分别对应的预测服装关键点、预测服装分割图像以及标准训练数据，对所述初始图像处理模型的模型参数进行调整，直到得到训练完成的目标图像处理模型。

根据本发明一个实施例提供了一种服装效果图的生成方法，该方法包括：

将获取到的待检测服装图像输入到预先训练完成的目标图像处理模型中；其中，所述目标图像处理模型包括特征提取网络、关键点定位网络和图像分割网络；

通过所述特征提取网络，对输入的所述待检测服装图像进行特征提取，得到图像特征；

通过所述关键点定位网络，基于所述特征提取网络输出的图像特征，输出所述待检测服装图像对应的目标服装关键点；

通过所述图像分割网络，基于所述特征提取网络输出的图像特征，输出所述待检测服装图像对应的目标服装分割图像；

基于所述目标服装关键点和所述目标服装分割图像，生成所述待检测服装图像对应的服装效果图。

根据本发明另一个实施例提供了一种图像处理模型的训练装置，该装置包括：