水下图像色彩还原方法和装置制造方法及图纸

本发明专利技术提供了一种水下图像色彩还原方法，该方法包括：S1：获取水下图像在RGB色彩格式下的各色彩通道的全局强度；S2：根据各色彩通道的全局强度的相对关系确定至少一个补偿通道，并结合水下图像的每个像素在补偿通道的像素值对该像素在对应的补偿通道进行增益补偿；其中，各色彩通道的全局强度由各色彩通道的像素值确定。与现有技术相比，本发明专利技术根据水下图像的各色彩通道的全局强度的相对关系来对水下图像中的每个像素按特定的步骤进行了色彩还原处理，使得水下画面的偏色程度相对统一，解决了现有技术中无法正确区分水下的前景物体和背景的水和/或使得前景物体的红色通道容易溢出等问题。溢出等问题。溢出等问题。

【技术实现步骤摘要】
水下图像色彩还原方法和装置


[0001]本申请涉及图像处理
，具体涉及一种水下图像的色彩还原方法、装置及计算机可读存储介质。

技术介绍

[0002]水下视觉不仅广泛运用于海洋探测、水下工程监测等的科学研究活动，也是广大摄影爱好者的重要拍摄素材来源，因此，获取真实的水下图像尤为重要。
[0003]市面上的数码相机在水下拍摄常常遇到偏色问题。由于不同波长的光在水中衰减率不同，其中波长较长的红光在水下传播衰减最为明显，水下拍摄画面中的物体普遍红色信息缺失，导致画面整体偏蓝或者偏绿。此时，相机的自动白平衡在此特殊的光照条件下容易失效，进一步导致画面色彩失真。
[0004]现有对水下拍摄的视频或图片的后期处理过程中，主要依靠人力对视频色彩进行手动调整，比如可以使用后期编辑软件对偏色视频或图片使用手动白平衡修正、通道混合器、色彩查找表去修正，在一定程度上可以还原出真实的水下色彩，但手动调整操作繁琐，而且当画面中的拍摄物体到相机的距离发生变化，或者拍摄的环境的水深发生较大变化时，画面偏色情况也会相应的改变，此时往往需要重新调色，费时费力。
[0005]针对上述问题，中国专利公开号为CN112348904A、专利技术名称为“水下图像及水下视频色彩还原方法和装置”公开了一种水下图像色彩还原方法，该方法通过将水下图像的各像素的由8位RGB像素值转换线性sRGB空间上进行调整处理，再将调整后的像素值转换至8位RGB像素值，然后将获取的8位RGB像素值与调整后的像素值进行融合，进而得到还原后的水下图像。但上述专利的技术方案存在以下缺陷：1、该方案需要计算一个权重值来表示衰减的程度。但是由于水下环境多变，计算衰减程度往往不准，进而无法正确区分水下的前景物体和背景的水。2、该方案主要通过计算通道增益去补偿水下的颜色，但是由于红色很少，所以算得红色增益较大，使得前景物体的红色通道容易溢出。
[0006]因此，有必要对现有的水下图像色彩还原方法进行改进。

技术实现思路

[0007]本专利技术的目的在于提供一种水下图像色彩还原方法、装置、电子设备及计算机可读存储介质，旨在解决现有一种水下图像的复原方法存在的缺陷。
[0008]第一方面，本专利技术提供了一种水下图像色彩还原方法，该方法包括：S1：获取水下图像在RGB色彩格式下的各色彩通道的全局强度；S2：根据各色彩通道的全局强度的相对关系确定至少一个补偿通道，并结合水下图像的每个像素在补偿通道的像素值对该像素在对应的补偿通道进行增益补偿；其中，各色彩通道的全局强度由各色彩通道的像素值确定。
[0009]第二方面，本专利技术提供了一种水下图像色彩还原装置，该装置包括：全局强度获取模块，用于获取水下图像在RGB色彩格式下的各色彩通道的全局强
度；补偿模块，用于根据各色彩通道的全局强度的相对关系确定至少一个补偿通道，并结合水下图像的每个像素在补偿通道的像素值对该像素在对应的补偿通道进行增益补偿；其中，各色彩通道的全局强度由各色彩通道的像素值确定。
[0010]第三方面，本专利技术提供了一种电子设备，包括：存储器，所述存储器存储有计算机程序；处理器，所述处理器用于执行所述计算机程序以实现上述的水下图像色彩还原方法。
[0011]第四方面，本专利技术提供了一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现上述的以实现上述的水下图像色彩还原方法。
[0012]与现有技术相比，本专利技术根据水下图像的各色彩通道的全局强度的相对关系和单个像素在各色彩通道的像素值来对水下图像中的每个像素按特定的步骤进行了色彩还原处理，使得水下画面的偏色程度相对统一，解决了现有技术中无法正确区分水下的前景物体和背景的水和/或使得前景物体的红色通道容易溢出等问题。
附图说明
[0013]图1是本专利技术实施例1中的水下图像色彩还原方法的流程图。
[0014]图2是本专利技术实施例1中的步骤S1的流程图。
[0015]图3是本专利技术实施例1中的步骤S2的流程图。
[0016]图4是本专利技术实施例2中的水下图像色彩还原装置的结构框图。
[0017]图5是本专利技术实施例3中的电子设备的结构框图。
具体实施方式
[0018]为了使本专利技术的目的、技术方案及有益效果更加清楚明白，以下结合附图及实施例，对本专利技术进行进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本专利技术，并不用于限定本专利技术。
[0019]为了说明本专利技术所述的技术方案，下面通过具体实施例来进行说明。
[0020]实施例1如图1所示，本实施例揭示了一种水下图像色彩还原方法，包括以下步骤：S0：检测水下图像是否为RGB色彩格式，如果不是则将其转换为RGB色彩格式。
[0021]水下图像是指拍摄装置（如相机或手机）的镜头在水中拍摄的照片或视频帧，水下图像可以是任意像素色彩格式（如BGR、RGB、YUV等），如果水下图像不是RGB色彩格式，则需要将其转换为RGB色彩格式。图像的不同色彩格式之间的转换为现有技术，本实施例中不再详细说明。
[0022]S1：获取水下图像在RGB色彩格式下的各色彩通道的全局强度。
[0023]本实施例的一个具体方案中，步骤S1包括以下子步骤。
[0024]S11：获取各像素在各色彩通道的像素值。
[0025]本实施例以8位RGB色彩空间为例进行说明，获取的水下图像在每个色彩通道中的像素值在0
‑
255之间。
[0026]S12：对各像素值进行归一化处理。
[0027]将获取的水下图像的各色彩通道的像素值除以255，使得各像素值变为0
‑
1之间，即完成了各像素值的归一化处理。
[0028]S13：根据归一化处理后的在各色彩通道的像素值确定各色彩通道的全局强度。
[0029]在本实施例中的一个具体方案中，是通过对水下图像的各像素在各色彩通道的像素值的平均值来确定各色彩通道的全局强度。具体的，假设水下图像有n个像素，r
i
、g
i
、b
i
分别代表第i个像素在红色通道、绿色通道和蓝色通道的像素值，则红色通道的像素平均值r
a
＝1/n*，绿色通道的像素平均值g
a
＝1/n*，蓝色通道的像素平均值b
a
＝1/n*。
[0030]在本实施例中的另一可选方案中，是通过对水下图像的各像素在各色彩通道的某个区段内的像素值的平均值来确定各色彩通道的全局强度。具体的，将水下图像的各像素在各色彩通道的像素值按从大到小进行排序，然后取位于中间的一定比例（如50%）的像素值，然后计算这些像素值的平均值。通过这种方式，可以降低水下图片在极端情况下（如曝光不足或水下图像部分区域被遮挡时）对各色彩通道的全局强度的影响。
[0031]在本实施例中的其他可选方案中，各色彩通道的全局强度还可以为对水下图像的各像素在各色本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种水下图像色彩还原方法，其特征在于，包括：S1：获取水下图像在RGB色彩格式下的各色彩通道的全局强度；S2：根据各色彩通道的全局强度的相对关系确定至少一个补偿通道，并结合水下图像的每个像素在补偿通道的像素值对该像素在对应的补偿通道进行增益补偿；其中，各色彩通道的全局强度由各色彩通道的像素值确定。2.根据权利要求1所述的水下图像色彩还原方法，其特征在于，所述各色彩通道的全局强度为：（1）水下图像的各像素在各色彩通道的像素值的平均值，或（2）对水下图像的各像素在各色彩通道的某个区段内的像素值的平均值，或（3）水下图像的各像素在各色彩通道的像素值的中位数。3.根据权利要求1所述的水下图像色彩还原方法，其特征在于，在所述步骤S1之前包括：S0：检测水下图像是否为RGB色彩格式，如果不是则将其转换为RGB色彩格式。4.根据权利要求1所述的水下图像色彩还原方法，其特征在于，所述步骤S1包括：S11：获取各像素在各色彩通道的像素值；S12：对各像素值进行归一化处理；S13：根据归一化处理后的在各色彩通道的像素值确定各色彩通道的全局强度。5.根据权利要求1所述的水下图像色彩还原方法，其特征在于，所述步骤S2包括：S21：判断蓝色通道的全局强度是否大于绿色通道的全局强度，如果是，则进入步骤S22，如果否则进入步骤S23;S22：对水下图像中的每个像素的绿色通道进行增益补偿；S23：对水下图像中的每个像素的蓝色通道进行增益补偿。6.根据权利要求5所述的水下图像色彩还原方法，其特征在于，所述步骤S22具体为：对于水下图像中的任一像素，其经增益补偿后的绿色通道的像素值g'的计算公式如下：g'=g+w
gb
*(b
a
‑
g
a
)*(1
‑
g)*b/*b
a
其中，g为该像素的在绿色通道的像素值，g
a
为水下图像的各像素在绿色通道的像素值的平均值、某个区段内的像素值的平均值或像素值的中位数，b为该像素的在蓝色通道的像素值，b
a
为水下图像的各像素在蓝色通道的像素值的平均值、某个区段内的像素值的平均值或像素值的中位数，w
gb
为经验值参数，表示根据该像素的蓝色通道强度去补偿绿色通道强度。7.根据权利要求5所述的水下图像色彩还原方法，其特征在于，所述步骤S23具体为：对于水下图像中的任一像素，其经增益补偿后的蓝色通道的像素值b'的计算公式如下：b'=b+w
bg
*(g
a
‑
b
a
)*(1
‑
b)*g/*g
a
其中，b为该像素的在蓝色通道的像素值，b
a
为水下图像的各像素在蓝色通道的像素值的平均值、某个区段内的像素值的平均值或像素值的中位数，g为该像素的在绿色通道的像素值，g
a
为水下图像的各像素在绿色通道的像素值的平均值、某个区段内的像素值的平均值或像素值的中位数，w
bg
...

【专利技术属性】
技术研发人员：苏坦，
申请(专利权)人：影石创新科技股份有限公司，
类型：发明
国别省市：