基于StableDiffusion的商品主图生成方法、装置、设备及介质与流程

本申请涉及人工智能，尤其涉及一种基于stable diffusion的商品主图生成方法、装置、设备及介质。背景技术：：：1、在stable diffusion技术兴起之前，人工智能的商品主图的技术实现形式往往是通过人工合成、按商品分类预制模板拼接来提升商品主图的人效，这种方式下设计的商品主图往往缺乏艺术性，风格较为单调统一，但能够一定程度提高生产效率。2、在stable diffusion技术兴起之后，出现了一些新兴的人工智能的商品主图生成方法，但效果上容易出现商品风格不统一和融合不自然等问题，导致还需要设计人员的二次介入进行修改，所以现有人工智能商品主图生成方法的生成质量和效率较低。技术实现思路1、本申请实施例的目的在于提出一种基于stable diffusion的商品主图生成方法、装置、设备及介质，以提高人工智能商品主图的生成质量和效率。2、为了解决上述技术问题，本申请实施例提供一种基于stable diffusion的商品主图生成方法，包括：3、获取待设计商品图，并去除所述待设计商品图的背景，得到目标待设计商品图；4、获取用户在画布上所添加的商品和设计元素，并生成所述目标待设计商品图对应的第一商品蒙版，以及生成所述商品和所述设计元素对应的第二商品蒙版和位置关系；5、获取所述用户所选择的场景，基于目标风格lora模型获取所述用户所选择的风格lora，并基于所述场景和所述风格lora生成对应的提示词和风格参数；6、基于所述目标待设计商品图、所述第一商品蒙版、所述第二商品蒙版、所述提示词、所述位置关系和所述风格参数进行图生图的重绘处理，得到初始商品主图；7、基于所述初始商品主图、所述第一商品蒙版和所述第二商品蒙版进行图生图的重绘处理，得到目标商品主图。8、为了解决上述技术问题，本申请实施例提供一种基于stable diffusion的商品主图生成装置，包括：9、待设计商品图获取单元，用于获取待设计商品图，并去除所述待设计商品图的背景，得到目标待设计商品图；10、商品蒙版生成单元，用于获取用户在画布上所添加的商品和设计元素，并生成所述目标待设计商品图对应的第一商品蒙版，以及生成所述商品和所述设计元素对应的第二商品蒙版和位置关系；11、风格参数生成单元，用于获取所述用户所选择的场景，基于目标风格lora模型获取所述用户所选择的风格lora，并基于所述场景和所述风格lora生成对应的提示词和风格参数；12、初始商品图生成单元，用于基于所述目标待设计商品图、所述第一商品蒙版、所述第二商品蒙版、所述提示词、所述位置关系和所述风格参数进行图生图的重绘处理，得到初始商品主图；13、目标商品主图生成单元，用于基于所述初始商品主图、所述第一商品蒙版和所述第二商品蒙版进行图生图的重绘处理，得到目标商品主图。14、为解决上述技术问题，本发明采用的一个技术方案是：提供一种计算机设备，包括，一个或多个处理器；存储器，用于存储一个或多个程序，使得一个或多个处理器实现上述任意一项所述的基于stable diffusion的商品主图生成方法。15、为解决上述技术问题，本发明采用的一个技术方案是：一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现上述任意一项所述的基于stable diffusion的商品主图生成方法。16、本发明实施例提供了一种基于stable diffusion的商品主图生成方法、装置、设备及介质。其中，方法包括：获取待设计商品图，并去除所述待设计商品图的背景，得到目标待设计商品图；获取用户在画布上所添加的商品和设计元素，并生成所述目标待设计商品图对应的第一商品蒙版，以及生成所述商品和所述设计元素对应的第二商品蒙版和位置关系；获取所述用户所选择的场景，基于目标风格lora模型获取所述用户所选择的风格lora，并基于所述场景和所述风格lora生成对应的提示词和风格参数；基于所述目标待设计商品图、所述第一商品蒙版、所述第二商品蒙版、所述提示词、所述位置关系和所述风格参数进行图生图的重绘处理，得到初始商品主图；基于所述初始商品主图、所述第一商品蒙版和所述第二商品蒙版进行图生图的重绘处理，得到目标商品主图。本发明实施例通过两阶段的生图处理，解决了商品风格不统一和融合不自然的问题，有利于提高人工智能商品主图的生成质量和效率。技术特征：1.一种基于stable diffusion的商品主图生成方法，其特征在于，包括：2.根据权利要求1所述的基于stable diffusion的商品主图生成方法，其特征在于，所述获取所述用户所选择的场景，基于目标风格lora模型获取所述用户所选择的风格lora，并基于所述场景和所述风格lora生成对应的提示词和风格参数之前，所述方法还包括：3.根据权利要求1所述的基于stable diffusion的商品主图生成方法，其特征在于，所述获取待设计商品图，并去除所述待设计商品图的背景，得到目标待设计商品图，包括：4.根据权利要求1所述的基于stable diffusion的商品主图生成方法，其特征在于，所述获取用户在画布上所添加的商品和设计元素，并生成所述目标待设计商品图对应的第一商品蒙版，以及生成所述商品和所述设计元素对应的第二商品蒙版和位置关系，包括：5.根据权利要求1所述的基于stable diffusion的商品主图生成方法，其特征在于，所述基于所述目标待设计商品图、所述第一商品蒙版、所述第二商品蒙版、所述提示词、所述位置关系和所述风格参数进行图生图的重绘处理，得到初始商品主图，包括：6.根据权利要求1至5任一项所述的基于stable diffusion的商品主图生成方法，其特征在于，所述基于所述初始商品主图、所述第一商品蒙版和所述第二商品蒙版进行图生图的重绘处理，得到目标商品主图，包括：7.根据权利要求1至5任一项所述的基于stable diffusion的商品主图生成方法，其特征在于，所述基于所述初始商品主图、所述第一商品蒙版和所述第二商品蒙版进行图生图的重绘处理，得到目标商品主图之后，所述方法还包括：8.一种基于stable diffusion的商品主图生成装置，其特征在于，包括：9.一种计算机设备，其特征在于，包括存储器和处理器，所述存储器中存储有计算机程序，所述处理器执行所述计算机程序时实现如权利要求1至7中任一项所述的基于stablediffusion的商品主图生成方法。10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至7中任一项所述的基于stablediffusion的商品主图生成方法。技术总结本申请涉及一种基于Stable Diffusion的商品主图生成方法、装置、设备及介质，其中方法包括：获取待设计商品图，去除待设计商品图的背景得到目标待设计商品图；获取用户添加的商品和设计元素，生成目标待设计商品图对应的第一商品蒙版，生成商品和设计元素对应的第二商品蒙版和位置关系；获取用户所选择的场景，基于目标风格LoRA模型获取用户所选择的风格LoRA，基于场景和风格LoRA生成对应的提示词和风格参数；基于目标待设计商品图、第一商品蒙版、第二商品蒙版、提示词、位置关系和风格参数进行图生图的重绘处理，得到初始商品主图；基于初始商品主图、第一商品蒙版和第二商品蒙版进行重绘处理，得到目标商品主图。本申请提高了人工智能商品主图的生成质量和效率。技术研发人员：赵汨,陈光炎受保护的技术使用者：深圳市和讯华谷信息技术有限公司技术研发日：技术公布日：2024/7/29