一种基于Lora模型训练的商品图换背景方法、装置、设备和介质

本发明提供一种基于Lora模型训练的商品图换背景方法、装置、设备和介质，方法包括：获取各种商品的第一商品图，得到训练数据；将每张所述第一商品图进行打标签，形成对应的标签文件；设置训练轮数、学习率以及训练图像像素，之后将训练数据与标签文件输入，进行模型训练，得到所需的背景Lora模型；将需要更换背景的第二商品图进行抠图，获取其中的商品主体蒙版；提取所述第二商品图的边缘信息；在图生图算法中加载所述背景Lora模型，将所述边缘信息、背景提示词以及商品主体蒙版输入至所述图生图算法中，生成商品替换图；所述背景提示词与所述设定格式背景标签对应，提高了制图的效率，降低了制图的成本。