[发明专利]一种单图像去反光的方法、装置及存储介质

摘要

本发明公开一种单图像去反光的方法、装置及存储介质，该方法包括：通过人工拍摄获取背景图像和对应的反射图像，并根据背景图像和反射图像的叠加，得到反光图像；将反光图像输入到预训练的VGG‑19网络进行超列特征提取，得到特征集合；将特征集合输入到预设的生成网络中，得到预测背景图和预测反射图；将预测背景图和背景图像输入到预设的鉴别网络，以计算得到鉴别网络的鉴别损失函数；通过多次迭代计算，直至生成网络的联合损失函数和鉴别损失函数均收敛，完成生成网络和鉴别网络的训练；选取多张反光图像进行去反光处理，以定量评估去反光效果。本发明能提取图像的高层次感官信息加入生成对抗网络的训练，从而有效解决单图像的去反光问题。