条件扩散概率模型训练方法及全向图像生成方法

    公开(公告)号:CN118096532A

    公开(公告)日:2024-05-28

    申请号:CN202410148281.9

    申请日:2024-02-01

    IPC分类号: G06T3/4053 G06T5/70

    摘要: 一种条件扩散概率模型训练方法及全向图像生成方法,涉及图像处理领域。在条件扩散概率模型训练方法,包括:获取低分辨率的全向训练图像,将全向训练图像输入至初始条件扩散概率模型,对全向训练图像进行离散小波变换,得到离散小波图像提取所述全向训练图像的纬度权重矩阵;对全向训练图像进行加噪处理,得到噪声图像;根据离散小波图像和所述纬度权重矩阵,对噪声图像进行去噪处理,得到去噪图像,对去噪图像进行逆离散小波变换,得到目标全向图像;根据目标全向图像计算损失值,根据损失值调整初始条件扩散概率模型,得到训练完成的条件扩散概率模型。采用本申请可以生成保留全局内容和几何结构信息的高分辨率全向图像。