基于自监督扩散模型的单幅图像反射去除
计算机视觉与模式识别
2024-12-31 v1 图像与视频处理
摘要
反射通常会降低通过透明表面拍摄的图像视觉质量，而反射去除方法缺乏成对真实样本。本文提出一种混合方法，将循环一致性与去噪扩散概率模型（DDPM）相结合，无需成对训练数据即可有效去除单幅图像中的反射。该方法引入了一个反射去除网络（RRN），利用DDPM建模分解过程并恢复透射图像，以及一个反射合成网络（RSN），通过基于非线性的注意力机制利用分离的组件重新合成输入图像。实验结果表明，该方法在SIR、基于闪光的反射去除（FRR）数据集以及新引入的博物馆反射去除（MRR）数据集上均有效，与最先进方法相比表现出优越性能。
引用
@article{arxiv.2412.20466,
title = {Single-image reflection removal via self-supervised diffusion models},
author = {Zhengyang Lu and Weifan Wang and Tianhao Guo and Feng Wang},
journal= {arXiv preprint arXiv:2412.20466},
year = {2024}
}