[发明专利]一种图像文件的生成方法、装置、设备和程序产品

说明书

本申请提供一种图像文件的生成方法、装置、设备和程序产品，涉及AI图像技术领域。一种图像文件的生成方法，包括：根据预设条件优化图像生成模型；获取训练图片和训练文本，所述训练文本包括图像主题信息；根据所述训练图片和所述训练文本对所述图像生成模型进行微调；根据输入的信息，通过已微调的所述图像生成模型生成图像文件。根据本申请的实施例，可保证图像生成模型在训练及图像文件生成的过程中的泛化性不受影响。

技术领域

本申请涉及AI图像技术领域，具体而言，涉及一种图像文件的生成方法、装置、设备和程序产品。

背景技术

随着通过AI生成素材内容的AIGC(AI Generated Content)技术的不断发展，人们对素材图像的个性化定制需求越来越旺盛。

传统平台提供的官方素材以icon为主，素材质量不高、丰富度低，并且外部优质的素材由于版权问题，无法做到二次分发并授权。而现阶段通过AIGC技术创作的素材，往往由于过于泛化的描述文本，导致AI创作出的结果存在图像中的元素过多且不可控制，以及图像的主体不聚焦的问题，无法满足实际诉求。

发明内容

根据本申请的一方面，提供一种图像文件的生成方法，包括：根据预设条件优化图像生成模型；获取训练图片和训练文本，所述训练文本包括图像主题信息；根据所述训练图片和所述训练文本对所述图像生成模型进行微调；根据输入的信息，通过已微调的所述图像生成模型生成图像文件。

根据一些实施例，根据预设条件优化图像生成模型，包括：通过所述图像生成模型进行编码；根据所述预设条件生成控制向量；根据所述控制向量进行解码，以控制所述图像生成模型的输出结果。

根据一些实施例，根据所述训练图片和所述训练文本对所述图像生成模型进行微调，包括：根据所述训练图片和所述训练文本对所述图像生成模型进行初始微调；根据所述训练图片对已经过所述初始微调的所述图像生成模型进行二次微调。

根据一些实施例，根据所述训练图片和所述文本信息对所述图像生成模型进行初始微调，包括：设置所述图像生成模型的损失函数；结合所述文本信息将所述训练图片输入所述图像生成模型，以获取先验图片；根据所述训练图片和所述先验图片更新所述损失函数。

根据一些实施例，根据所述训练图片对已经过初始微调的所述图像生成模型进行二次微调，包括：获取所述训练图片对应的下采样图片；根据所述下采样图片和所述训练图片生成下采样-训练图片组合；通过所述下采样-训练图片组合更新所述图像生成模型。

根据一些实施例，根据输入的信息，通过已微调的所述图像生成模型生成图像文件，包括：根据所述预设条件输入图像主题文本；根据所述图像主题文本通过已微调的所述图像生成模型生成所述图像文件。

根据一些实施例，所述图像主题文本包括与所述训练文本的图像主题信息匹配的关键字信息。

根据本申请的一方面，提供一种图像文件的生成装置，包括：数据采集模块，用于获取训练图片和训练文本；获取图像主题文本；数据处理模块，用于设置并根据预设条件优化图像生成模型；根据所述训练图片和所述训练文本对所述图像生成模型进行初始微调及二次微调；根据所述图像主题文本通过已微调的所述图像生成模型生成图像文件；输出模块，输出所述图像文件。

根据本申请的一方面，提供一种电子设备，包括：一个或多个处理器；存储装置，用于存储一个或多个程序；当所述一个或多个程序被所述一个或多个处理器执行，使得一个或多个处理器实现如前述的方法。

根据本申请的一方面，提供一种计算机程序产品，包括计算机程序或指令，所述计算机程序或指令被处理器执行时实现如前述的方法。

根据本申请的实施例，通过对图像生成模型的微调，保持模型生成的素材图像的风格的一致性，在输入的描述信息不够具象化的情况下生成符合诉求的相似度适合的图像文件，提升了创作效率。