一种实现元素替换的图像生成方法、装置及电子设备

本发明实施例提供了一种实现元素替换的图像生成方法、装置及电子设备，方法包括：根据给定的主体图和背景图，将所述主体图和背景图输入到训练的图像生成模型中，所述主体图中存在预先选取的需要替换的主体元素；所述图像生成模型中的图像编码模块获取主体图的第一特征向量，背景图的第二特征向量和主体元素的掩码向量；将所述第一特征向量、第二特征向量和掩码向量融合，生成融合的向量，所述图像生成模型中的图像生成模块根据所述向量和随机的高斯噪声图像进行迭代处理，输出目标图像，所述目标图像中为主体元素位于所述背景图中的图像；上述方法解决了现有技术中处理速度慢、效率低且编辑质量需要复杂后处理的问题。