本说明书一个或多个实施例涉及机器学习,尤其涉及一种宠物微调模型的训练方法、装置和电子设备。
背景技术:
1、在数字互动娱乐与虚拟现实技术迅猛发展的今天,数字宠物已成为许多人生活中的一部分。数字宠物,也称为虚拟宠物或电子宠物,是一种虚拟程序,旨在模拟真实宠物的不同方面,如动作、情感反应以及与用户的互动。
2、随着人们生活水平的提高,宠物行业迅速增长,宠物主人对于宠物越来越关注。宠物主人也越来越有兴趣参与数字宠物应用程序的互动,切尔对于社交共享和在不同社交媒体平台展示其数字宠物的需求不断增长。特别是,用户希望他们的数字宠物能够更独特和具有较高的辨识度。但是,现有技术的数字宠物尚无法满足用户的上述需求。
技术实现思路
1、有鉴于此,本说明书一个或多个实施例提供一种宠物微调模型的训练方法、装置和电子设备,以提高模型生成的数字宠物的生成质量。
2、为实现上述目的,本说明书一个或多个实施例提供技术方案如下:
3、根据本说明书实施例的第一方面,提供了一种宠物微调模型的训练方法,所述方法包括:
4、获取包括多个训练图像的训练图像集,每个训练图像中包括宠物,且所述多个训练图像中包括的宠物是同一个宠物;
5、在对宠物微调模型进行训练的过程中,获取所述宠物微调模型输出的预测图像,并基于所述预测图像与对应的所述训练图像之间的差异,调整所述宠物微调模型的参数;所述训练图像作为所述宠物微调模型的训练标签;
6、在所述预测图像满足预设的质量条件的情况下,停止所述宠物微调模型的训练;所述预设的质量条件包括:所述预测图像中的宠物与对应的所述训练图像中的宠物之间的相似度达到第一阈值。
7、根据本说明书实施例的第二方面,提供了一种数字宠物图像的生成方法,所述方法包括:
8、接收数字宠物图像的生成请求,所述生成请求中包括:宠物标识;
9、获取宠物基础模型、以及所述宠物标识对应的宠物微调模型;所述宠物微调模型是根据本说明书任一实施例所述的方法训练得到;
10、基于宠物基础模型和宠物微调模型,生成所述宠物标识对应宠物的数字宠物图像。
11、根据本说明书实施例的第三方面,提供了一种宠物微调模型的训练装置,所述装置包括:
12、图像获取模块,用于获取包括多个训练图像的训练图像集,每个训练图像中包括宠物,且所述多个训练图像中包括的宠物是同一个宠物;
13、模型训练模块,用于在对宠物微调模型进行训练的过程中,获取所述宠物微调模型输出的预测图像,并基于所述预测图像与对应的所述训练图像之间的差异,调整所述宠物微调模型的参数;所述训练图像作为所述宠物微调模型的训练标签;
14、模型监控模块,用于在所述预测图像满足预设的质量条件的情况下,停止所述宠物微调模型的训练;所述预设的质量条件包括:所述预测图像中的宠物与对应的所述训练图像中的宠物之间的相似度达到第一阈值。
15、根据本说明书实施例的第四方面,提供了一种数字宠物图像的生成装置,包括:
16、请求接收模块,用于接收数字宠物图像的生成请求,所述生成请求中包括:宠物标识;
17、模型获取模块,用于获取宠物基础模型、以及所述宠物标识对应的宠物微调模型;所述宠物微调模型是根据本说明书任一实施例所述的方法训练得到;
18、生成处理模块,用于基于所述宠物基础模型和宠物微调模型,生成所述宠物标识对应宠物的数字宠物图像。
19、根据本说明书实施例的第五方面,提供了一种电子设备,包括:
20、处理器;
21、用于存储处理器可执行指令的存储器;
22、其中,所述处理器通过运行所述可执行指令以实现本说明书任一实施例所述的方法。
23、根据本说明书实施例的第六方面,提供了一种计算机可读存储介质,其上存储有计算机指令,该指令被处理器执行时实现本说明书任一实施例所述的方法。
24、根据本说明书实施例的第七方面,提供了一种计算机程序产品,包括计算机程序,所述计算机程序被处理器执行时实现本说明书任一实施例所述的方法。
25、本说明书实施例的宠物微调模型的训练方法、装置和电子设备,通过在模型训练过程中,通过训练图像中的宠物来监控宠物微调模型的训练,在模型输出的预测图像与训练图像中的宠物之间的相似度达到第一阈值的情况下,才停止宠物微调模型的训练,使得能够尽可能的训练宠物微调模型输出与训练图像中的宠物相似度较高的宠物,提高模型输出的图像与用户宠物的相似度。
1.一种宠物微调模型的训练方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,
3.根据权利要求1所述的方法,其特征在于,
4.根据权利要求3所述的方法,其特征在于,所述将所述训练图像的宠物主体与图像背景进行分割处理,得到所述宠物主体对应的子图像,包括:
5.根据权利要求1至4任一所述的方法,其特征在于,所述方法还包括:
6.根据权利要求1所述的方法,其特征在于,
7.根据权利要求1所述的方法,其特征在于,所述方法还包括:
8.一种数字宠物图像的生成方法,其特征在于,所述方法包括:
9.一种宠物微调模型的训练装置,其特征在于,所述装置包括:
10.一种数字宠物图像的生成装置,其特征在于,所述装置包括:
11.一种电子设备,其特征在于,包括:
12.一种计算机可读存储介质,其上存储有计算机指令,其特征在于,该指令被处理器执行时实现如权利要求1至8中任一项所述方法的步骤。
13.一种计算机程序产品,包括计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至8中任一项所述方法的步骤。
