基于RGB图像和扩散先验的偏振图像生成方法及装置

本申请公开了一种基于RGB图像和扩散先验的偏振图像生成方法及装置，方法包括：获取并归一化RGB图像的像素值，得到原始图像参数；将原始图像参数输入预先训练的偏振图像生成模型中，输出RGB图像对应的扩散先验信息，扩散先验信息包括编码的线性偏振角和线性偏振度；预先训练的偏振图像生成模型是对大型扩散模型的预训练权重进行微调得到的；根据编码的线性偏振角和线性偏振度，计算原始图像对应的预设不同偏振角度的光通过强度；对预设不同偏振角度的光通过强度进行可视化处理，得到多个偏振图像。因此，采用本申请实施例，通过模型可输出扩散先验信息，该信息在无需依赖三维资产的情况下，仅通过单一RGB图像可生成逼真效果又有物理准确性的偏振图像。