Deep learning based methods have achieved significant success in the task of single image reflection removal (SIRR). However, the majority of these methods are focused on High-Definition/Standard-Definition (HD/SD) images, while ignoring higher resolution images such as Ultra-High-Definition (UHD) images. With the increasing prevalence of UHD images captured by modern devices, in this paper, we aim to address the problem of UHD SIRR. Specifically, we first synthesize two large-scale UHD datasets, UHDRR4K and UHDRR8K. The UHDRR4K dataset consists of $2,999$ and $168$ quadruplets of images for training and testing respectively, and the UHDRR8K dataset contains $1,014$ and $105$ quadruplets. To the best of our knowledge, these two datasets are the first largest-scale UHD datasets for SIRR. Then, we conduct a comprehensive evaluation of six state-of-the-art SIRR methods using the proposed datasets. Based on the results, we provide detailed discussions regarding the strengths and limitations of these methods when applied to UHD images. Finally, we present a transformer-based architecture named RRFormer for reflection removal. RRFormer comprises three modules, namely the Prepossessing Embedding Module, Self-attention Feature Extraction Module, and Multi-scale Spatial Feature Extraction Module. These modules extract hypercolumn features, global and partial attention features, and multi-scale spatial features, respectively. To ensure effective training, we utilize three terms in our loss function: pixel loss, feature loss, and adversarial loss. We demonstrate through experimental results that RRFormer achieves state-of-the-art performance on both the non-UHD dataset and our proposed UHDRR datasets. The code and datasets are publicly available at https://github.com/Liar-zzy/Benchmarking-Ultra-High-Definition-Single-Image-Reflection-Removal.
翻译:基于深度学习的单图像反射去除方法已取得显著成功。然而,这些方法大多聚焦于高清/标清图像,忽略了如超高清(UHD)等更高分辨率的图像。随着现代设备捕获的超高清图像日益普及,本文旨在解决超高清单图像反射去除问题。具体而言,我们首先合成两个大规模超高清数据集UHDRR4K和UHDRR8K。其中,UHDRR4K数据集包含2,999组训练图像四元组和168组测试图像四元组,UHDRR8K数据集包含1,014组训练图像四元组和105组测试图像四元组。据我们所知,这两个数据集是迄今为止针对单图像反射去除的最大规模超高清数据集。随后,我们利用所提数据集对六种最先进的单图像反射去除方法进行了全面评估。基于实验结果,我们详细讨论了这些方法在处理超高清图像时的优势与局限。最后,我们提出一种名为RRFormer的基于Transformer的反射去除架构。RRFormer包含三个模块:预处理嵌入模块、自注意力特征提取模块和多尺度空间特征提取模块,分别用于提取超列特征、全局与局部注意力特征以及多尺度空间特征。为确保有效训练,损失函数采用像素损失、特征损失和对抗损失三项。实验结果表明,RRFormer在非超高清数据集和所提超高清数据集上均达到了最先进性能。代码与数据集已开源至https://github.com/Liar-zzy/Benchmarking-Ultra-High-Definition-Single-Image-Reflection-Removal。