Object Detection Using Structure-Preserving Wavelet Pyramid Reflection Removal Network

Images shot through glass will produce reflections. However, these reflections will cause the objects in the images to be unrecognizable. Hence, to improve the recognition rate, the reflections in the images should be removed. Reflection removal has been widely used in the field of deep learning. Although these methods have good results, they all assume that reflection removal is performed in a specific situation and provide their own datasets for research, such as strong reflections in some local areas, and this limitation will lead to the inability to effectively remove the reflections in the real world, which will affect the recognition of objects. To address this issue, we propose a novel structure-preserving wavelet pyramid reflection removal network (SpWPRRNet) to achieve effective background structure preservation and reflection removal to further improve the object recognition rate in images. After wavelet decomposition, we put the high- and low-frequency images of each level into reflection layer removal and detail enhancement subnetwork (RDSn) and structure preservation subnetwork (SPSn), respectively, and then use structure-level fusion (SLF) and inverse stationary wavelet transform (ISWT) to restore clean images recursively. In addition, to further separate the reflection layer and the transmission layer, we propose the reflection layer information transmission (RLIT), through which the reflection layer features of the high-frequency images can be extracted to help the SPSn to effectively separate these two layers to achieve the results of reflection removal and improve object recognition rate. The experimental results indicate that the proposed method can greatly improve the object recognition rate in images.