构建5300对像素级对齐的去反射数据集,推动真实场景下单图像去反射研究。
OpenRR-5k: A Large-Scale Benchmark for Reflection Removal in the Wild
- 构建5300对高精度像素对齐图像对,涵盖多样真实场景。
- 训练集5000张,验证集300张,另有100张无真值测试图评估实际性能。
- 提供完整代码与数据集,支持多种主流评估指标验证效果。
去除反射是计算机视觉中的关键任务,广泛应用于摄影和图像增强。然而,现有方法受限于缺乏大规模、高质量且多样的数据集。本文提出一个用于单图像反射去除(SIRR)的新基准。我们构建了一个包含5,300对高质量像素对齐图像对的大规模数据集,每对包含一张带反射图像及其对应的干净版本。其中5,000张用于训练,300张用于验证,另有100张无真值的真实世界测试图像以评估方法在实际场景中的表现。所有图像对均在像素级别精确对齐,确保监督信号准确。数据集覆盖多种光照条件、物体类型和反射模式,按训练、验证、测试集划分,便于全面评估。为验证数据集有效性,我们基于U-Net训练模型,并使用PSNR、SSIM、LPIPS、DISTS和NIQE五种常用指标进行评估。相关数据集与代码将开源至https://github.com/caijie0620/OpenRR-5k,以推动该领域研究。
原文摘要 · Abstract (English)
Removing reflections is a crucial task in computer vision, with significant applications in photography and image enhancement. Nevertheless, existing methods are constrained by the absence of large-scale, high-quality, and diverse datasets. In this paper, we present a novel benchmark for Single Image Reflection Removal (SIRR). We have developed a large-scale dataset containing 5,300 high-quality, pixel-aligned image pairs, each consisting of a reflection image and its corresponding clean version. Specifically, the dataset is divided into two parts: 5,000 images are used for training, and 300 images are used for validation. Additionally, we have included 100 real-world testing images without ground truth (GT) to further evaluate the practical performance of reflection removal methods. All image pairs are precisely aligned at the pixel level to guarantee accurate supervision. The dataset encompasses a broad spectrum of real-world scenarios, featuring various lighting conditions, object types, and reflection patterns, and is segmented into training, validation, and test sets to facilitate thorough evaluation. To validate the usefulness of our dataset, we train a U-Net-based model and evaluate it using five widely-used metrics, including PSNR, SSIM, LPIPS, DISTS, and NIQE. We will release both the dataset and the code on https://github.com/caijie0620/OpenRR-5k to facilitate future research in this field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。