HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Single Image Reflection Removal Exploiting Misaligned Training Data and Network Enhancements

Kaixuan Wei; Jiaolong Yang; Ying Fu; David Wipf; Hua Huang

Single Image Reflection Removal Exploiting Misaligned Training Data and Network Enhancements

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

Vandermode/ERRNet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
reflection-removal-on-real20ERRNet
PSNR: 22.89
SSIM: 0.803
reflection-removal-on-sir-2-objectsERRNet
PSNR: 24.87
SSIM: 0.896
reflection-removal-on-sir-2-postcardERRNet
PSNR: 22.04
SSIM: 0.876
reflection-removal-on-sir-2-wildERRNet
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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Single Image Reflection Removal Exploiting Misaligned Training Data and Network Enhancements | Papers | HyperAI