一种基于领域自适应的扩散模型逆转图像重建方法和装置

    公开(公告)号:CN118887138A

    公开(公告)日:2024-11-01

    申请号:CN202410846620.0

    申请日:2024-06-27

    Abstract: 本发明涉及一种基于领域自适应的扩散模型逆转图像重建方法和装置。该方法包括:获取待处理的源域高质量图像和目标域低质量图像作为初始输入图像;利用初始输入图像,使用预训练的扩散模型预测正向扩散过程中下一时刻的隐码,不断迭代直至得到每个时刻的隐码,至最终时刻时得到的隐码为符合高斯分布的噪声,在迭代的每一步都采用域修正网络在每个时刻对扩散模型预测的隐码进行修正;进而使用预训练的扩散模型预测逆向扩散过程中上一时刻的隐码,不断迭代直至得到重建图像。本发明在没有监督的情况下自发学习源域中高质量图像的有效表示并将其迁移到目标域中的低质量图像中,同时实现了对高质量和低质量两种图像的重建和编辑。

Patent Agency Ranking