基于水扩散物理模型的水下图像增强方法

本发明提供一种基于水扩散物理模型的水下图像增强方法，涉及水下图像增强技术领域，该方法包括：S101、输入水下图像的局部块到卷积神经网络，预测水体的衰减系数矩阵和扩散矩阵；S102、基于水体的衰减系数矩阵和扩散矩阵构建水扩散控制方程；S103、基于水扩散控制方程对水下图像在频域执行多尺度扩散，获得重建图像；S104、通过光谱透射率模型恢复重建图像的颜色通道，获得增强图像；S105、输出增强图像。所述方法通过估计光在水下的衰减和散射系数，实现更符合光在水下物理扩散特性的图像增强，有效避免颜色失真和图像模糊，提升增强结果的物理可解释性和可靠性。