一种基于残差双注意力网络的水下图像增强方法
摘要:
本发明公开了一种基于残差双注意力网络的水下图像增强方法,具体按照以下步骤实施:步骤1、对输入的图像进行预处理;步骤2、将预处理的图像映射到R通道、G通道、B通道,分别对每个颜色通道进行单独的卷积操作;步骤3、使用基于残差双注意力的编解码器的深度网络架构对图像进行处理,编码器用于提取图像特征,减少空间维度,增大感受野;解码器用于逐步恢复恢复图像的细节信息和空间维度;利用预处理的图像对残差双注意力网络模型进行训练,得到训练好的残差双注意力网络模型;步骤4、将模糊的低质量的水下图像输入所述训练好的残差双注意力网络模型中,得到清晰的水下图像。解决了现有技术中存在的水下图像存在的颜色偏差的问题。
0/0