本发明涉及图像处理,尤其是一种水下图像增强方法、装置、设备及介质。
背景技术:
1、水中的悬浮粒子和大量浮游生物对光的散射作用,以及水对不同波长的光的吸收作用造成了水下图像颜色衰减、对比度低、模糊等问题,对海洋的开发造成了严重的影响。
2、水下成像技术的提升可以为海洋的开发提供更多的有用信息,方法是对获取的水下信息通过算法进行处理。其中一种方法为基于深度学习的图像增强,然而现有的基于深度学习的水下图像增强算法一般在成对的水下图像和陆上真实图像上进行监督学习,但是在现实环境中,难以获得成对的水下图像和其对应的陆上真实图像,通过补光等手段消除散射和衰减的影响,拍摄不受水下环境影响的图像,是一个极其难以完成的任务。因为水流和设备抖动以及人工操作等原因,难以保证两幅图像是对着完全相同的场景拍摄的,研究者们几乎不可能采集到严格符合标准的对应陆上真实图像。大多数的研究者都是基于水下成像模型将陆上图像水下合成为水下图像制作成对的数据集,但是当前研究中合成的水下图像数据集难以广泛地反应水下成像的真实特征。实际应用过程中,会使有太亮或者太暗的水下图像过度曝光或者曝光不足,同时也会使图像的细节损失。
技术实现思路
1、有鉴于此,本发明实施例提供一种水下图像增强方法、装置、设备及介质,能够有效进行水下图像增强,提升水下图像质量。
2、一方面,本发明的实施例提供了一种水下图像增强方法,包括:
3、获取原始水下图像,对原始水下图像进行对比度拉伸,获得第一图像;第一图像包括红通道、绿通道和蓝通道;
4、对第一图像进行图像分解,获得各个通道的基础层和细节层;
5、根据绿通道的细节层或蓝通道的细节层,结合红通道的基础层进行红通道补偿,获得补偿红通道;
6、对补偿红通道、绿通道和蓝通道进行合并,获得第二图像;
7、对第二图像进行色彩校正,得到目标水下图像。
8、可选地,对原始水下图像进行对比度拉伸,获得第一图像,包括:
9、通过对比度受限自适应直方图均衡方法,对原始水下图像的各通道进行对比度拉伸,分别获得红通道、绿通道和蓝通道。
10、可选地,对第一图像进行图像分解这一步骤前,方法还包括:
11、利用双边滤波对第一图像进行去噪预处理,获得去噪后的第一图像。
12、可选地,对第一图像进行图像分解,获得各个通道的基础层和细节层,包括:
13、对第一图像进行均值滤波,获得各个通道的基础层;
14、根据第一图像中各个通道与各个通道对应的基础层的差值,获得各个通道的细节层。
15、可选地,根据绿通道的细节层或蓝通道的细节层,结合红通道的基础层进行红通道补偿,包括:
16、基于绿通道和蓝通道,依次确定绿通道均值和蓝通道均值;
17、当绿通道均值大于蓝通道均值,通过绿通道的细节层结合红通道的基础层进行红通道补偿;
18、当蓝通道均值大于绿通道均值,通过蓝通道的细节层结合红通道的基础层进行红通道补偿。
19、可选地,对第二图像进行色彩校正,得到目标水下图像,包括:
20、利用灰度世界算法对第二图像进行色彩校正,得到目标水下图像。
21、可选地,利用灰度世界算法对第二图像进行色彩校正,得到目标水下图像,包括:
22、基于补偿红通道、绿通道和蓝通道,依次确定补偿红通道均值、绿通道均值和蓝通道均值;
23、对补偿红通道均值、绿通道均值和蓝通道均值进行平均处理,获得灰度均值;
24、根据灰度均值与补偿红通道均值的比值,结合预设权重系数,获得第一补偿系数;并根据灰度均值与绿通道均值和蓝通道均值的比值,获得第二补偿系数;
25、通过第一补偿系数对第二图像的补偿红通道进行第一补偿处理,并通过第二补偿系数对第二图像的绿通道和蓝通道进行第二补偿处理,得到目标水下图像。
26、另一方面,本发明的实施例提供了一种水下图像增强装置,包括:
27、第一模块,用于获取原始水下图像,对原始水下图像进行对比度拉伸,获得第一图像;第一图像包括红通道、绿通道和蓝通道;
28、第二模块,用于对第一图像进行图像分解,获得各个通道的基础层和细节层;
29、第三模块,用于根据绿通道的细节层或蓝通道的细节层,结合红通道的基础层进行红通道补偿,获得补偿红通道;
30、第四模块,用于对补偿红通道、绿通道和蓝通道进行合并,获得第二图像;
31、第五模块,用于对第二图像进行色彩校正,得到目标水下图像。
32、可选地,第二模块进行处理前,装置还包括:
33、第六模块,用于利用双边滤波对第一图像进行去噪预处理,获得去噪后的第一图像。
34、另一方面,本发明的实施例提供了一种电子设备,包括处理器以及存储器;
35、存储器用于存储程序;
36、处理器执行程序实现如前面的方法。
37、另一方面,本发明的实施例提供了一种计算机可读存储介质,存储介质存储有程序,程序被处理器执行实现如前面的方法。
38、本发明实施例还公开了一种计算机程序产品或计算机程序,该计算机程序产品或计算机程序包括计算机指令,该计算机指令存储在计算机可读存储介质中。计算机设备的处理器可以从计算机可读存储介质读取该计算机指令,处理器执行该计算机指令,使得该计算机设备执行前面的方法。
39、本发明实施例首先获取原始水下图像,对原始水下图像进行对比度拉伸,获得第一图像;第一图像包括红通道、绿通道和蓝通道;对第一图像进行图像分解,获得各个通道的基础层和细节层;根据绿通道的细节层或蓝通道的细节层,结合红通道的基础层进行红通道补偿,获得补偿红通道;对补偿红通道、绿通道和蓝通道进行合并,获得第二图像;对第二图像进行色彩校正,得到目标水下图像。本发明实施例通过对水下图像进行对比度拉伸以及图像分解和红通道补偿实现颜色差异性衰减补偿,从而提升水下图像的质量,本发明能够在不依靠水下成像系统的情况下对水下图像进行处理从而提升水下图像的质量。
1.一种水下图像增强方法,其特征在于,包括:
2.根据权利要求1所述的一种水下图像增强方法,其特征在于,所述对所述原始水下图像进行对比度拉伸,获得第一图像,包括:
3.根据权利要求1所述的一种水下图像增强方法,其特征在于,所述对所述第一图像进行图像分解这一步骤前,所述方法还包括:
4.根据权利要求1所述的一种水下图像增强方法,其特征在于,所述对所述第一图像进行图像分解,获得各个通道的基础层和细节层,包括:
5.根据权利要求1所述的一种水下图像增强方法,其特征在于,所述根据所述绿通道的细节层或所述蓝通道的细节层,结合所述红通道的基础层进行红通道补偿,包括:
6.根据权利要求1所述的一种水下图像增强方法,其特征在于,所述对所述第二图像进行色彩校正,得到目标水下图像,包括:
7.根据权利要求6所述的一种水下图像增强方法,其特征在于,所述利用灰度世界算法对所述第二图像进行色彩校正,得到目标水下图像,包括:
8.一种水下图像增强装置,其特征在于,包括:
9.一种电子设备,其特征在于,包括处理器以及存储器;
10.一种计算机可读存储介质,其特征在于,所述存储介质存储有程序,所述程序被处理器执行实现如权利要求1至7中任一项所述的方法。