Nov, 2023

基于条件扩散模型的高质量 HDR 去鬼影

TL;DR从低动态范围(LDR)图像中恢复高动态范围(HDR)图像是通过现有的深度神经网络(DNN)技术实现的。但是,基于 DNN 的方法在 LDR 图像存在饱和度和大动作时仍然会生成虚影伪影,这限制了其在真实世界场景中的应用潜力。为了解决这个挑战,我们将 HDR 去伪影问题转化为基于图像生成的问题,利用 LDR 特征作为扩散模型的条件,包括特征条件生成器和噪声预测器。特征条件生成器利用注意力和域特征对齐(DFA)层来转换中间特征,以避免虚影伪影。通过学习得到的特征作为条件,噪声预测器利用随机迭代去噪过程来通过控制采样过程生成 HDR 图像。此外,为了减轻 LDR 图像饱和问题引起的语义混淆,我们设计了一个滑动窗口噪声估计器,以块为单位进行平滑噪声采样。另外,提出了一个图像空间损失来避免估计的 HDR 结果的色彩失真。我们通过在 HDR 成像的基准数据集上进行实证评估,结果表明我们的方法在性能上达到了最先进水平,并且对真实世界图像具有良好的泛化能力。