基于RAW图像的端到端雨痕去除

在这项工作中，我们处理了RAW图像的雨痕去除问题。一般方法是首先将RAW数据处理成RGB图像，然后用RGB图像去除雨痕。实际上，RAW图像中的原始雨息会受到图像信号处理（ISP）流水线的影响，包括非线性算法、意外噪声、伪影等。在处理成RGB格式之前直接在RAW数据中去除雨痕会获得更多益处。为了解决这个问题，我们提出了一种联合雨去除和RAW处理的解决方案，以从有雨的RAW图像中获得干净的颜色图像。具体而言，我们通过将颜色雨痕转换到RAW空间来生成有雨的RAW数据，并设计了简单但高效的RAW处理算法来合成有雨和干净的颜色图像。有雨的颜色图像被用作参考以辅助颜色校正。不同的主干网络表明，与专注于颜色图像的其他几种最先进去雨方法相比，我们的方法取得了更好的结果。此外，所提出的网络对超出我们选定RAW数据集的其他相机具有良好的泛化能力。最后，我们给出了在不同ISP流水线处理的图像上的测试结果，以表明与基于颜色图像的方法相比，我们模型的泛化性能更好。