基于多条件控制扩散模型的图像生成方法和电子设备

本发明涉及一种基于多条件控制扩散模型的图像生成方法和电子设备，包括如下步骤：步骤S1，获取图像训练集，针对其中每个图像数据，提取图像的多个条件，构建训练样本；步骤S2，基于所述训练样本，以最小化变分自编码器预测得到的预测条件与实际条件之间的区别为目标，对变分自编码器进行训练；步骤S3，随机选取一个或多个条件作为子集，利用所述变分自编码器计算子集对应的编码特征，基于所述编码特征以及加噪后的图像数据，以最小化预测的噪声和真实的噪声之间的差别为目标，对扩散模型进行训练；步骤S4，利用训练后的扩散模型实现图像生成。与现有技术相比，本发明具有有效管理和整合多个输入条件、有效管理条件冲突等优点。