训练试衣模型的方法、生成试衣图像的方法及相关装置与流程

技术特征：
1.一种训练试衣模型的方法，其特征在于，试衣网络包括衣服编码网络和解析融合网络；所述方法包括：获取训练集，所述训练集包括多个训练数据，所述训练数据包括衣服图像和真实试衣图像，所述真实试衣图像包括模特穿有所述衣服图像中对应衣服的图像；采用人体解析算法对所述真实试衣图像进行解析，得到初步人体解析图；采用衣服编码网络对所述衣服图像进行下采样编码，得到衣服特征图；将所述初步人体解析图和所述衣服特征图输入所述解析融合网络进行融合解析，得到预测人体解析图；采用损失函数计算所述训练集中各所述真实试衣图像对应的初步人体解析图和预测人体解析图之间的差异，其中，损失函数包括用于约束所述衣服图像中的衣服和所述初步人体解析图中人体之间位置关系的约束函数；根据所述差异，对所述试衣网络进行迭代训练，直到所述试衣网络收敛，得到所述试衣模型。2.根据权利要求1所述的方法，其特征在于，所述解析融合网络包括编码器、融合模块和解码器，所述将所述初步人体解析图和所述衣服特征图输入所述解析融合网络进行融合，得到预测人体解析图，包括：将所述初步人体解析图输入所述编码器进行下采样编码，得到人体特征图；将所述人体特征图和所述衣服特征图输入所述融合模块进行融合，得到融合有衣服风格信息的融合特征图；将所述融合特征图输入所述解码器进行上采样解码，得到所述预测人体解析图。3.根据权利要求2所述的方法，其特征在于，所述将所述人体特征图和所述衣服特征图输入所述融合模块进行融合，得到融合有衣服风格信息的融合特征图，包括：将所述人体特征图中的像素和所述衣服特征图中的像素，通过对应位置相乘，获得所述融合特征图。4.根据权利要求1-3任意一项所述的方法，其特征在于，所述约束函数包括至少一个预设区域约束函数，所述预设区域约束函数用于约束预设衣服区域和预设人体区域之间的距离。5.根据权利要求4所述的方法，其特征在于，还包括：采用人体关键点算法对所述真实试衣图像进行关键点检测，得到模特人体关键点；采用衣服关键点算法对所述衣服图像进行关键点检测，得到衣服关键点；将所述模特人体关键点和所述衣服关键点代入所述约束函数进行计算，以满足所述约束函数所约束的距离关系。6.根据权利要求5所述的方法，其特征在于，所述至少一个预设区域约束函数包括以下不等式：
(|f
2-c4| |f
5-c5|)<ε3其中，ε1为左手臂和左袖之间的距离约束值，f2和f4分别为所述左手臂的关键点坐标，c4和c6分别为所述左袖的关键点坐标，ε2为右手臂和右袖之间的距离约束值，f5和f7分别为所述右手臂的关键点坐标，c5和c8分别为所述右袖的关键点坐标，ε3为肩膀和衣服肩部之间的距离约束值，ε4为臀部和衣服下摆之间的距离约束值，f8和f
11
分别为所述臀部的关键点坐标，c
12
和c
13
分别为所述衣服下摆的关键点坐标。7.根据权利要求1所述的方法，其特征在于，在所述采用人体解析算法对所述真实试衣图像进行解析，得到初步人体解析图之前，还包括：对所述真实试衣图像进行填充处理，以使所述真实试衣图像的分辨率满足预设比例。8.一种生成试衣图像的方法，其特征在于，包括：获取待试衣服图像和用户图像；采用人体解析算法对所述用户图像进行解析，得到所述用户的初步人体解析图；将所述待试衣服图像和所述用户的初步人体解析图输入试衣模型，得到所述用户的新人体解析图，其中，所述试衣模型采用如权利要求1-7中任意一项训练试衣模型的方法训练得到；将所述用户的新人体解析图的区域类别进行隐藏处理，得到所述试衣图像。9.一种计算机设备，其特征在于，包括：至少一个处理器，以及与所述至少一个处理器通信连接的存储器，其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-8任一项所述的方法。10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机可执行指令，所述计算机可执行指令用于使计算机设备执行如权利要求1-8任一项所述的方法。

技术总结
本申请实施例涉及图像处理技术领域，公开了一种训练试衣模型的方法、生成试衣图像的方法及相关装置，试衣网络包括衣服编码网络和解析融合网络，训练集包括多个训练数据，各训练数据包括衣服图像和真实试衣图像。采用人体解析算法对真实试衣图像进行解析，得到初步人体解析图。采用衣服编码网络对衣服图像进行下采样编码，得到衣服特征图。采用初步人体解析图和衣服特征图作为数据，结合损失函数，对解析融合网络进行训练，得到试衣模型。其中，损失函数包括用于约束试穿衣服和人体之间位置关系的约束函数。在约束函数的作用下，试穿衣服能够准确定位，和模特结合贴切，另外，还能够使得衣服能够保持自身的风格特征，不受模特原有衣服特征的影响。服特征的影响。服特征的影响。


技术研发人员：陈仿雄
受保护的技术使用者：深圳数联天下智能科技有限公司
技术研发日：2022.03.22
技术公布日：2022/7/29