图像生成模型的训练方法、装置、电子设备及存储介质

    公开(公告)号:CN119919755A

    公开(公告)日:2025-05-02

    申请号:CN202411997082.1

    申请日:2024-12-31

    Inventor: 达铖 张迪

    Abstract: 本公开关于图像生成模型的训练方法、装置、电子设备及存储介质,方法包括:获取样本图像、包括加噪网络和去噪网络的图像生成模型和待训练一致性模型;基于加噪网络和当前采样时间步对样本图像进行加噪得到第一样本噪声图像;基于去噪网络和预设数量个单位时间间隔对第一样本噪声图像进行去噪得到第二样本噪声图像;将第一样本噪声图像和第二样本噪声图像输入待训练一致性模型,得到第一样本预测图像和第二样本预测图像;根据第一样本预测图像、第二样本预测图像和与当前采样时间步负相关的动态权重所确定的目标损失数据,对待训练一致性模型进行训练,得到目标图像生成模型。本公开可以得到能直接生成图像的模型并可以提升模型的图像生成质量。

Patent Agency Ranking