本申请属于图像处理领域,具体涉及一种水下图像增强系统及方法。
背景技术:
1、相较于陆地资源探索已充分,海洋资源探索仍初级。海洋资源包括石油、天然气和渔业资源等,对人类发展重要。但由于海洋环境复杂多变,探索开发面临巨大困难。水下图像增强处理可更好了解海洋资源分布、开发难度、潜力和确定开发范围和方案。然而,水下图像常受多种因素影响,如光线传播、悬浮颗粒浓度等,出现模糊、颜色偏差和低对比度问题,需专业技术和设备解决。
2、水下图像增强技术能够提升能见度、减小色差和提高对比度三个方面,有利于观察、分析水下场景和提高计算机视觉任务性能。传统方法利用cnn和gan技术实现水下图像增强,只考虑局部信息提取。虽然有一些工作引入视觉注意力机制提取一定的全局信息,但增强效果还有待提高。
技术实现思路
1、为解决上述背景中的技术问题,本申请提出一种基于交叉注意力机制的水下图像增强系统及方法。通过块间注意力模块获得局部信息;同时,通过块间注意力机制来获得全局信息进行建模,从而实现更完整的图像增强效果。
2、为实现上述目的,本申请提供了一种水下图像增强系统,包括:交叉注意力机制模块、动态特征增强模块和特征融合模块;
3、所述交叉注意力机制模块用于提取水下图像中的全局信息和局部信息并结合,得到图像特征;
4、所述动态特征增强模块用于提高所述图像特征的表达能力和区别能力;
5、所述特征融合模块用于提高所述图像特征的重建能力;所述特征融合模块还用于进一步增强所述图像特征的区别能力,完成水下图像的增强。
6、优选的,在所述交叉注意力机制模块获取所述图像特征前,应当将待提取的水下图像经过一个重置大小归一化层和一个镜像填充层进行图像大小处理;同时进行线性映射,用以提取所述图像特征。
7、优选的,所述交叉注意力机制模块包括:块内注意力机制和块间注意力机制;所述交叉注意力机制模块的工作流程包括:
8、由所述块内注意力机制提取特征图的局部信息;再以所述局部信息为输出,输入到所述块间注意力机制中,提取所述特征图的全局信息;之后,将所述局部信息和所述全局信息进行融合,得到所述图像特征。
9、优选的,所述动态特征增强模块包括:两个可变形卷积,所述可变性卷积为在普通卷积的卷积核位置引入一个可学习的偏移量。
10、优选的,所述动态特征增强模块的工作流程包括:所述卷积核通过在输入特征图的采样点时发生偏移,集中于感兴趣的区域或者目标,用以加强所述图像特征的表达能力和区别能力。
11、优选的,所述特征融合模块是一种用于融合多个分支的通道注意力机制,由卷积层、池化层以及softmax操作组成。
12、优选的,所述特征融合模块的工作流程包括:使用通道注意力来融合所述动态特征增强模块的输出分支和所述交叉注意力机制模块的输出分支;以融合原始特征信息和经过多层交叉注意力机制的特征信息,以提高所述图像特征的重建能力;并进一步增强所述图像特征的区别能力。
13、本申请还提供了一种水下图像增强方法,步骤包括:
14、提取水下图像中的全局信息和局部信息并结合,得到图像特征;
15、提高所述图像特征的表达能力和区别能力;
16、提高所述图像特征的重建能力;并进一步增强所述图像特征的区别能力,完成水下图像的增强。
17、与现有技术相比,本申请的有益效果如下:
18、本申请能够恢复水下图片中的真实颜色;并且在海洋探测和资源勘探方面,本申请可以帮助科学家和工程师更好地理解海洋生物和地质环境,从而更好地探索和利用海洋资源。在水下安全监测和救援方面,本申请可以帮助救援人员快速准确地发现和定位事故现场、失踪人员和潜水器等。此外,在水下文物保护和考古发掘方面,本申请可以帮助文物保护者和考古学家更好地了解水下文化遗产的状况和历史背景,从而更好地保护和挖掘这些宝贵的文化遗产。
1.一种水下图像增强系统,其特征在于,包括:交叉注意力机制模块、动态特征增强模块和特征融合模块;
2.根据权利要求1所述的水下图像增强系统,其特征在于,在所述交叉注意力机制模块获取所述图像特征前,将待提取的水下图像经过一个重置大小归一化层和一个镜像填充层进行图像大小处理;同时进行线性映射,用以提取所述图像特征。
3.根据权利要求1所述的水下图像增强系统,其特征在于,所述交叉注意力机制模块包括:块内注意力机制和块间注意力机制;所述交叉注意力机制模块的工作流程包括:
4.根据权利要求1所述的水下图像增强系统,其特征在于,所述动态特征增强模块包括:两个可变形卷积,所述可变性卷积为在普通卷积的卷积核位置引入一个可学习的偏移量。
5.根据权利要求4所述的水下图像增强系统,其特征在于,所述动态特征增强模块的工作流程包括:所述卷积核通过在输入特征图的采样点时发生偏移,集中于感兴趣的区域或者目标,用以加强所述图像特征的表达能力和区别能力。
6.根据权利要求1所述的水下图像增强系统,其特征在于,所述特征融合模块是一种用于融合多个分支的通道注意力机制,由卷积层、池化层以及softmax操作组成。
7.根据权利要求6所述的水下图像增强系统,其特征在于,所述特征融合模块的工作流程包括:使用通道注意力来融合所述动态特征增强模块的输出分支和所述交叉注意力机制模块的输出分支;以融合原始特征信息和经过多层交叉注意力机制的特征信息,以提高所述图像特征的重建能力;并进一步增强所述图像特征的区别能力。
8.一种水下图像增强方法,其特征在于,步骤包括: