Abstract:To address the issue of image quality decline caused by existing reflection removal algorithms when handling complex scenes, this study proposes a color-aware dual-channel reflection removal algorithm. First, a background color generator is designed to accurately predict the background color information of an image, provide background support for the basic reflection removal network, and generate preliminary reflection removal results. Subsequently, a dual-channel reflection removal network is proposed to further optimize these preliminary results. Additionally, the algorithm designs a sparse Transformer module, a channel attention module, and a feature fusion module within the dual-channel reflection removal network, thereby enhancing the precision and effect of reflection removal. Experimental results demonstrate that this method performs excellently on the RRID and Flash datasets, effectively removing reflected light and significantly enhancing image realism.