一种基于卷积神经网络和注意力机制的光场深度预测方法

摘要

本发明涉及一种基于卷积神经网络和注意力机制的光场深度预测方法，包括：获取光场图像并对其进行预处理，生成光场图像集；构建光场深度预测模型，该模型包括EPI学习模块、注意力模块和特征融合模块；将光场图像集分别输入EPI学习模块和注意力模块，分别获取光场图像的EPI信息以及各个图像权重；分别将光场图像EPI信息以及各个图像的权重输入特征融合模块，获得光场深度预测结果。与现有技术相比，本发明具有预测精度高、实用性好等优点。