用于图片创作的模型训练方法及装置与流程

本说明书一个或多个实施例涉及终端，尤其涉及一种用于图片创作的模型训练方法及装置。

背景技术：

1、在当今的数字艺术和设计领域，造型师和美术人员面临着越来越高的创作要求和紧迫的时间压力，传统的美术创作依赖于个人经验和手工技能，需要耗费大量的时间和精力。

2、随着人工智能技术的发展，尤其是深度学习和生成对抗网络(generativeadversarial network，gan)的普及，图片生成技术为美术创作提供了新的可能性。相关技术可以基于由用户提供的提示词，由预训练的大语言模型(large language model，llm)模型进行图片推理，来生成与提示词相对应的图片。但如此生成的图片同质化比较严重，用户为了使生成的图片具有自身的风格特征，需要不断的调整提示词，严重影响了图片创作的效率。

技术实现思路

1、有鉴于此，本说明书一个或多个实施例提供技术方案如下：

2、根据本说明书一个或多个实施例的第一方面，提出了一种用于图片创作的模型训练方法，包括：

3、向用户输出用于模型训练的交互界面；

4、获取用户在所述交互界面中输入的包含所述用户的自定义图像特征的样本图片，以及模型训练参数；

5、响应于用户触发的开启训练操作，得到用于生成包含所述自定义图像特征的图片的用户模型，所述用户模型是基于所述样本图片和所述模型训练参数对预训练完成的基础模型进行微调训练得到的。

6、根据本说明书一个或多个实施例的第二方面，提出了一种用于图片创作的模型训练方法，包括：

7、界面输出模块，用于向用户输出用于模型训练的交互界面；

8、用户输入模块，用于获取用户在所述交互界面中输入的包含所述用户的自定义图像特征的样本图片，以及模型训练参数；

9、模型训练模块，用于响应于用户触发的开启训练操作，得到用于生成包含所述自定义图像特征的图片的用户模型，所述用户模型是基于所述样本图片和所述模型训练参数对预训练完成的基础模型进行微调训练得到的。

10、根据本说明书一个或多个实施例的第三方面，提出了一种电子设备，包括：处理器；用于存储处理器可执行指令的存储器；其中，所述处理器通过运行所述可执行指令以实现如第一方面所述方法的步骤。

11、根据本说明书一个或多个实施例的第四方面，提出了一种计算机可读存储介质，其上存储有计算机指令，该指令被处理器执行时实现如第一方面所述方法的步骤。

12、根据本说明书一个或多个实施例的第五方面，提出了一种计算机程序产品包括计算机程序/指令，该计算机程序/指令被处理器执行时实现如第一方面所述方法的步骤。

13、由上述实施例可知，本说明书可以将样本图片作为训练样本图片来对预训练的基础模型进行微调训练，通过模型训练参数对微调训练过程进行引导或约束，使微调训练后得到的用户模型能够准确学习到样本图片包含的用户的自定义图像特征，从而在调用该用户模型进行图片推理时，可以使生成的图片具有与样本图片相同的用户的自定义图像特征，满足用户自身的需求，使得用户通过简单的操作即可得到所需要的用户模型，来生成所需要的图片，进而提升图片创作的效率和作品的独特性。

技术特征：

1.一种用于图片创作的模型训练方法，其特征在于，应用于与图片创作平台对应的客户端，所述方法包括：

2.根据权利要求1所述的方法，其特征在于，所述基础模型为用于生成包含预设的多个维度的图像特征的图片的模型；所述模型训练参数包括用户从所述预设的多个维度的图像特征中选择的目标维度的图像特征；相应的，所述用户模型用于生成包含所述目标维度的自定义图像特征的图片。

3.根据权利要求1所述的方法，其特征在于，所述模型训练参数包括用户从所述图片创作平台提供的预训练完成的多个基础模型中选定的目标基础模型；相应的，所述用户模型是基于样本图片和所述模型训练参数对预训练完成的目标基础模型进行微调训练得到的。

4.根据权利要求3所述的方法，其特征在于，所述多个基础模型分别对应不同的图片生成类型。

5.根据权利要求1所述的方法，其特征在于，所述模型训练参数包括用户为所述用户模型设置的触发词；所述触发词用于指示调用所述用户模型进行图片推理来生成图片。

6.根据权利要求1-5中任一所述的方法，其特征在于，所述模型训练参数还包括以下示出的至少一项或多项的组合：

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

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

9.根据权利要求8所述的方法，其特征在于，所述调用微调训练完成的所述用户模型对所述提示词进行图片推理，以生成包含所述自定义图像特征的图片，包括：

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

11.根据权利要求10所述的方法，其特征在于，所述调用所述用户模型基于所述强度值对所述提示词进行图片推理，包括：

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

13.一种电子设备，其特征在于，包括：处理器；用于存储处理器可执行指令的存储器；其中，所述处理器通过运行所述可执行指令以实现如权利要求1-12中任一项所述方法的步骤。

14.一种计算机可读存储介质，其特征在于，其上存储有计算机指令，该指令被处理器执行时实现如权利要求1-12中任一项所述方法的步骤。

15.一种计算机程序产品，其特征在于，包括计算机程序/指令，该计算机程序/指令被处理器执行时实现如权利要求1-12中任一项所述方法的步骤。

技术总结本说明书一个或多个实施例提供一种用于图片创作的模型训练方法及装置，其中，所述方法包括：向用户输出用于模型训练的交互界面；获取用户在所述交互界面中输入的包含所述用户的自定义图像特征的样本图片，以及模型训练参数；响应于用户触发的开启训练操作，得到用于生成包含所述自定义图像特征的图片的用户模型，所述用户模型是基于所述样本图片和所述模型训练参数对预训练完成的基础模型进行微调训练得到的。技术研发人员：王华辉,范文龄,张乙申,曹祎冰,唐继正,王哲,张一丰,边洪冰,贾晨曦受保护的技术使用者：神力视界（深圳）文化科技有限公司技术研发日：技术公布日：2024/10/17