一种电商生图模型的训练和优化方法

本发明公开了一种电商生图模型的训练和优化方法，属于图像生成领域，包括：获取电商的图像数据，对所述图像数据进行标注，得到类别先验；构建电商生图模型和可视先验嵌入模块，通过随机噪声注入和扩散式去噪的训练方式对主干扩散模型进行训练；对所述图像数据进行编码，得到潜在变量，将类别先验结合潜在变量输入所述主干扩散模型，得到初步图像；通过可视先验嵌入模块对初步图像进行压缩重建，得到视觉先验信息，将视觉先验信息和类别先验拼接后再结合输入条件输入所述主干扩散模型，得到优化后的电商展示图像。通过上述技术方案，能够有效提升电商生图模型的生成图像的质量和细节。