一种基于扩散模型的条件引导图像翻译方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于扩散模型的条件引导图像翻译方法
申请号:CN202410829403
申请日期:2024-06-25
公开号:CN118822841A
公开日期:2024-10-22
类型:发明专利
摘要
本发明公开了一种基于扩散模型的条件引导图像翻译方法,所述方法提出一种结合扩散模型与条件生成对抗网络的两阶段图像翻译模型,阶段一预训练ResAttNet1提取深度特征,利用深度特征图中包含的丰富语义信息,作为条件生成对抗网络的条件信息,引导条件生成对抗网络完成图像翻译。阶段二利用训练好的ResAttNet1和参数随机初始化的ResAttNet2分别提取全局特征信息和样本级深度特征,采用条件生成对抗网络与扩散模型联合训练的策略,使用一个轻量的扩散模型细化深度特征,最终构建一个训练稳定、生成图像保真度好、采样速率高的图像翻译网络。该方法能够提高条件信息的质量和准确性,有效提升CGAN的图像翻译性能。
技术关键词
条件生成对抗网络 图像翻译方法 可见光图像 注意力 传播算法 样本 参数 翻译模型 噪声方差 通道 阶段 因子 数学 语义 策略