科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种深度学习单图像去反射方法、装置、电子设备和存储介质,该方法包括：构造训练集；利用网络中的编码器对训练集中带反射图像I进行特征提取,得到n个不同尺度的编码特征图；根据n个不同尺度的编码特征图,在网络的解码过程中,生成各尺度的解码特征图、传输层图像特征图和反射区域导向掩膜图,并得到最终预测的传输层图像；利用训练集中真实的传输层图像T对网络进行监督,指导网络预测的传输层图像更接近真实传输层图像,达到图像去反射的目的；同时对生成的掩膜图进行监督,限制掩膜图的取值范围,指导网络生成更精确掩膜图,引导网络准确预测传输层图像。本发明能够很好地对单张反射图像完成去反射任务。
- 单位
