一种基于弱监督学习的图像镜面高光去除算法
Weakly Supervised Specular Highlight Removal with Only Highlight Images
-
摘要: 高质量图像是计算机视觉任务的基础,但实际生活中高光的出现会覆盖物体表面的纹理和颜色信息,导致图像质量显著下降。目前用于高光去除的深度学习方法往往需要大量高光-无高光配对图像进行监督,而高光图像对应的无高光版本存在收集和处理困难的问题。本文提出一种基于弱监督学习的图像镜面高光去除算法,旨在仅使用高光图像完成训练且达到很好的高光去除效果。首先,利用稀疏非负矩阵分解(NMF)方法估计图像的高光区域,并从无高光区域裁剪出无高光的参考图像。然后,将两者输入到联合训练的高光生成、高光消除和图像重建模块,协同优化各模块功能。总体采用循环生成对抗网络(CycleGAN)架构训练网络并最终生成无高光图像。选取自然图像数据集SHIQ和LIME进行实验,实验结果表明,所提方法能够有效去除镜面高光,并且在性能上对比现有的弱监督学习方法有较大提升。Abstract: High-quality images are the fuels for computer vision tasks, but the existence of highlights in real life will cover the texture and color information of the object surface, resulting in substantial image quality degradation. Current deep learning based methods for highlight removal often require a large number of highlight-highlight-free paired images for supervision, while the corresponding highlight-free versions of the highlight images are hard to collect and process. This paper proposes an image specular highlight removal algorithm based on weakly supervised learning, which uses only highlight images in training and achieves promising highlight removal performance. First, a sparse non-negative matrix factorization (NMF) method is used to estimate the highlight region of the image, and a highlight-free reference image is cropped from the highlight-free region. Then, the unpaired highlight and highlight-free images are fed into the jointly trained highlight generation, highlight removal, and image reconstruction modules to optimize the functionality of the respective module. The overall network is trained with a Cyclic Generative Adversarial Network (CycleGAN) architecture and eventually generates a highlight-free image. The natural image datasets SHIQ and LIME are selected for experiments. The experimental results show that the proposed method can effectively remove specular highlights, and the performance is substantially improved compared with the existing weakly supervised learning methods.