[发明专利]一种基于迁移学习的光场图像雨条纹检测与去除方法

摘要

本发明公开一种基于迁移学习的光场图像雨条纹检测与去除方法，该方法包括深度图计算模块、雨条纹检测模块和雨水去除模块。本发明首先利用深度图计算模块得到深度图，然后通过雨条纹检测模块检测出合成数据雨条纹图并利用高斯过程模块自监督地检测出真实场景数据雨条纹图，最后将带雨3DEPI体积块、得到的深度图和提取的雨条纹图串联之后，输入3D递归生成对抗网络到中进行雨水的去除，如此反复训练，直到得到高质量的无雨图。相比于同类型工作，本发明利用迁移学习构建自监督网络，能够更准确地提取真实场景的雨条纹并得到高质量的无雨图，具有良好的泛化能力。