位置感知的单图像反射去除

本文提出一种新颖的基于位置感知深度学习的单图像反射去除方法。我们的网络具有一个反射检测模块，以多尺度拉普拉斯特征作为输入回归概率性反射置信度图。该概率图指示一个区域是以反射为主还是以透射为主，并被用作线索以在网络预测反射层与透射层时控制特征流。我们将网络设计为循环网络，以在每次迭代中逐步精炼反射去除结果。其新颖之处在于利用拉普拉斯核参数来强化强反射的边界。这有利于强反射检测并大幅改善反射去除结果的质量。大量实验验证了所提方法相较最先进方法的优越性能。我们的代码与预训练模型可在 https://github.com/zdlarr/Location-aware-SIRR 找到。