[发明专利]模型训练方法、装置和存储介质

权利要求书

1.一种模型训练方法，包括：

获取属于第一图像类别的第一样本和属于第二图像类别的第二样本；

将所述第一样本依次经过第一阶段图像转化模型和第一阶段图像逆向转化模型，得到第一样本单阶段恢复图像；

将所述第二样本依次经过所述第一阶段图像逆向转化模型和所述第一阶段图像转化模型，得到第二样本单阶段恢复图像；

按照所述第一样本与所述第一样本单阶段恢复图像的差异，及所述第二样本与所述第二样本单阶段恢复图像的差异，调整所述第一阶段图像转化模型和所述第一阶段图像逆向转化模型，直至满足训练停止条件时结束训练；

将所述第一样本依次经过所述第一阶段图像转化模型和第二阶段图像转化模型，根据所述第一阶段图像转化模型输出的第一样本一阶转化图像，和所述第二阶段图像转化模型输出的第一样本二阶转化图像，得到与所述第一样本对应、且属于所述第二图像类别的第一样本转化图像；

将所述第一样本转化图像依次经过所述第一阶段图像逆向转化模型和第二阶段图像逆向转化模型，根据所述第一阶段图像逆向转化模型输出的第一样本一阶恢复图像，和所述第二阶段图像逆向转化模型输出的第一样本二阶恢复图像，得到与所述第一样本对应、且属于所述第一图像类别的第一样本恢复图像；

将所述第二样本依次经过所述第一阶段图像逆向转化模型和第二阶段图像逆向转化模型，根据所述第一阶段图像逆向转化模型输出的第二样本一阶转化图像，和所述第二阶段图像逆向转化模型输出的第二样本二阶转化图像，得到与所述第二样本对应、且属于所述第一图像类别的第二样本转化图像；

将所述第二样本转化图像依次经过所述第一阶段图像转化模型和第二阶段图像转化模型，根据所述第一阶段图像转化模型输出的第二样本一阶恢复图像，和所述第二阶段图像转化模型输出的第二样本二阶恢复图像，得到与所述第二样本对应、且属于所述第二图像类别的第二样本恢复图像；

按照所述第一样本与所述第一样本恢复图像的差异，及所述第二样本与所述第二样本恢复图像的差异，调整所述第二阶段图像转化模型和所述第二阶段图像逆向转化模型，直至满足训练停止条件时结束训练。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

获取所述第一样本经过所述第一阶段图像转化模型后，由所述第一阶段图像转化模型输出的第一样本单阶段转化图像；

获取所述第二样本经过所述第一阶段图像逆向转化模型后，由所述第一阶段图像逆向转化模型输出的第二样本单阶段转化图像；

将所述第一样本和所述第二样本单阶段转化图像分别输入第一阶段图像逆向转化鉴别模型，分别得到所述第一样本的鉴别置信度和所述第二样本单阶段转化图像的鉴别置信度；

将所述第二样本和所述第一样本单阶段转化图像分别输入第一阶段图像转化鉴别模型，分别得到所述第二样本的鉴别置信度和所述第一样本单阶段转化图像的鉴别置信度；

所述按照所述第一样本与所述第一样本单阶段恢复图像的差异，及所述第二样本与所述第二样本单阶段恢复图像的差异，调整所述第一阶段图像转化模型和所述第一阶段图像逆向转化模型，直至满足训练停止条件时结束训练，包括：

按照最大化所述第一样本的鉴别置信度和所述第二样本的鉴别置信度的方向、最小化所述第二样本单阶段转化图像的鉴别置信度、所述第一样本单阶段转化图像的鉴别置信度、所述第一样本与所述第一样本单阶段恢复图像的差异，及所述第二样本与所述第二样本单阶段恢复图像的差异的方向，调整所述第一阶段图像转化鉴别模型、所述第一阶段图像逆向转化鉴别模型、所述第一阶段图像转化模型和所述第一阶段图像逆向转化模型，直至满足训练停止条件时结束训练。