中文

基于循环平移CNN的不确定性引导多尺度残差学习用于单图像去雨

计算机视觉与模式识别 2019-06-27 v1 机器学习 图像与视频处理 机器学习

摘要

单图像去雨是一个极具挑战性的问题,因为含雨图像可能包含尺寸、方向和密度各异的雨纹。先前的方法尝试通过利用某些先验信息从单张图像中去除雨纹。这些方法的一个主要局限在于它们未考虑图像中雨滴的位置信息。所提出的不确定性引导多尺度残差学习 (UMRL) 网络试图通过在不同尺度上学习雨内容,并利用它们来估计最终的去雨输出,以解决此问题。此外,我们引入了一种技术,基于对估计的置信度度量来引导网络学习网络权重。更进一步,我们引入了一种基于循环平移概念的新型训练和测试流程,以提升最终的去雨性能。在合成和真实数据集上的大量实验表明,所提出的方法相较于近期最先进的方法取得了显著改进。代码可在以下链接获取:https://github.com/rajeevyasarla/UMRL--using-Cycle-Spinning

关键词

引用

@article{arxiv.1906.11129,
  title  = {Uncertainty Guided Multi-Scale Residual Learning-using a Cycle Spinning CNN for Single Image De-Raining},
  author = {Rajeev Yasarla and Vishal M. Patel},
  journal= {arXiv preprint arXiv:1906.11129},
  year   = {2019}
}

备注

The IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2019