Single-Image Reflection Removal Via a Two-Stage Background Recovery Process

In this project, we propose a novel two-stage DNN based reflection removal algorithm. In the first stage, we include a new feature reduction term in the loss function when training the network. Due to its strong reflection suppression ability, the reflection components in the image can be more effectively suppressed. However, it will also attenuate the gradient values of the background image. For recovering the background, in the second stage, we first estimate a reflection gradient confidence map based on the initial estimation result and use it to identify the strong background gradients. Then we use a generative adversarial network to reconstruct the background image from its gradients. Experimental results show that the proposed two-stage approach can give superior performance compared with the state-of-the-art DNN based methods.

Downloadable resource: