基于环境光感知的端到端水下图像复原方法

摘要

本发明公开了一种基于环境光感知的端到端水下图像复原方法，主要解决现有技术在处理水下图像时色偏校正和清晰化处理效果欠佳的问题。其方案是：利用Pytorch框架分别构建环境光感知网络和复原主体网络，并分别构建这两个网络的训练集B和C；采用自适应矩估计算法分别利用B和C训练环境光感知网络和复原主体网络，将待处理的图像Ic输入训练好的环境光感知网络，输出环境光值Ac；将Ac和Ic输入训练好的复原主体网络，输出清晰图像Jc。本发明提高了不同退化程度水下图像的对比度，能有效校正色偏，且峰值信噪比、结构相似性、色差公式、无参考图像空间质量评估和水下彩色图像质量评价均优于现有技术，可用于水下图像的清晰化处理。