Multi-dimensional perception-guided iterative reflection removal network with deep features for painting images
摘要
To address the issue of image quality degradation caused by glass cover reflections during the digitization of Thangka paintings, this paper introduces an innovative iterative prediction network (MPGINet) guided by reflection perception and based on multi-dimensional deep features. The proposed network enhances the robustness of reflection removal by leveraging deep multi-dimensional features from the initially predicted reflection layer, transmission layer, and the original input image. This is further augmented by the strategic integration of frequency-domain information separation and mask-guided image inpainting. The proposed model employs a two-stage iterative architecture. In the first stage, it integrates the U-Net framework with the Squeeze-and-Excitation module, enhanced by a residual mechanism, to refine the prediction of the reflection layer. In the second stage, the network utilizes the novel Deep Feature Pyramid Network (DFPN), which excels in capturing the fine-grained texture features of Thangka paintings. The DFPN effectively fuses high- and low-frequency features under mask constraints, enabling precise restoration of the transmission layer. This dual-stage approach ensures a comprehensive and detailed recovery of the original artwork’s visual fidelity. Experimental results show that on the Thangka datasets, the PSNR and SSIM of MPGINet reach 28.90 dB and 0.962 respectively, an increase of 1.88 dB and 0.027 compared with the existing state-of-the-art (SOTA) methods. In natural scene datasets, the proposed method achieves comparable results with SOTA methods, verifying its generalization ability.