Single Image Reflection Removal Exploiting Misaligned Training Data and Network Enhancements
Single Image Reflection Removal Exploiting Misaligned Training Data and Network Enhancements
Kaixuan Wei Jiaolong Yang Ying Fu David Wipf Hua Huang

Abstract
Removing undesirable reflections from a single image captured through a glass window is of practical importance to visual computing systems. Although state-of-the-art methods can obtain decent results in certain situations, performance declines significantly when tackling more general real-world cases. These failures stem from the intrinsic difficulty of single image reflection removal -- the fundamental ill-posedness of the problem, and the insufficiency of densely-labeled training data needed for resolving this ambiguity within learning-based neural network pipelines. In this paper, we address these issues by exploiting targeted network enhancements and the novel use of misaligned data. For the former, we augment a baseline network architecture by embedding context encoding modules that are capable of leveraging high-level contextual clues to reduce indeterminacy within areas containing strong reflections. For the latter, we introduce an alignment-invariant loss function that facilitates exploiting misaligned real-world training data that is much easier to collect. Experimental results collectively show that our method outperforms the state-of-the-art with aligned data, and that significant improvements are possible when using additional misaligned data.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| reflection-removal-on-real20 | ERRNet | PSNR: 22.89 SSIM: 0.803 |
| reflection-removal-on-sir-2-objects | ERRNet | PSNR: 24.87 SSIM: 0.896 |
| reflection-removal-on-sir-2-postcard | ERRNet | PSNR: 22.04 SSIM: 0.876 |
| reflection-removal-on-sir-2-wild | ERRNet | PSNR: 24.25 SSIM: 0.853 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.