4 个月前

用于单图像反射去除的可逆解耦网络

用于单图像反射去除的可逆解耦网络

摘要

近期基于深度学习的单图像反射去除方法取得了显著进展,这主要归因于两个方面:1)利用预训练的识别特征作为输入;2)设计双流交互网络。然而,根据信息瓶颈原理,高层次的语义线索在逐层传播过程中往往会受到压缩或被丢弃。此外,双流网络中的交互模式在不同层次上遵循固定模式,限制了整体性能。为了解决这些局限性,我们提出了一种新的架构——可逆解耦网络(Reversible Decoupling Network, RDNet),该网络采用可逆编码器来保护有价值的信息,并在前向传播过程中灵活地解耦传输相关和反射相关的特征。此外,我们定制了一个传输率感知的提示生成器,以动态校准特征,进一步提升性能。大量实验表明,RDNet 在五个广泛使用的基准数据集上优于现有的最先进方法。RDNet 在 NTIRE 2025 单图像反射去除挑战赛中,在保真度和感知比较方面均取得了最佳性能。我们的代码已发布在 https://github.com/lime-j/RDNet

代码仓库

lime-j/RDNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
reflection-removal-on-natureRDNet
PSNR: 26.21
SSIM: 0.842
reflection-removal-on-natureZhu et al.
PSNR: 26.04
SSIM: 0.846
reflection-removal-on-real20RDNet
PSNR: 25.58
SSIM: 0.846
reflection-removal-on-sir-2-objectsRDNet
PSNR: 26.78
SSIM: 0.921
reflection-removal-on-sir-2-objectsZhu et al.
SSIM: 0.931
reflection-removal-on-sir-2-postcardRDNet
PSNR: 26.33
SSIM: 0.922
reflection-removal-on-sir-2-wildRDNet
PSNR: 27.7
SSIM: 0.915

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于单图像反射去除的可逆解耦网络 | 论文 | HyperAI超神经