白平衡计算方法技术

本申请的一种白平衡计算方法，包括：获取初始图像，并根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比；若所述第一占比大于预设的第一阈值，计算所述干扰色对应的类白区的第二占比；若所述第二占比大于预设的第二阈值，对所述初始图像中的所述干扰色进行抠除处理，得到目标图像；对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数，从而提高白平衡增益参数计算的准确性

【技术实现步骤摘要】
白平衡计算方法、装置、设备和介质


[0001]本申请涉及图像处理
，例如涉及一种白平衡计算方法
、
装置
、
设备和介质
。

技术介绍

[0002]目前相机采取的
AWB(Automatic White Balance
，自动白平衡
)
算法大多是灰度世界和白点统计的方法，是采取统计出
RAW
格式
(
原始格式
)
的图像中类白区的灰点来计算出
AWB
的增益，然而在实际应用中，偏白或偏黄的肤色
、
绿色等某些干扰色的落点也会落在类白区，甚至会被当作中色温的白点计算，从而导致最后计算出来的
AWB
增益参数不精确，使得根据
AWB
增益参数增益后的图像出现偏色的情况
。

技术实现思路

[0003]本申请目的在于：提供一种白平衡计算方法
、
装置
、
设备和介质，其能够提高白平衡增益参数计算的准确性
。
[0004]为达到上述目的，本申请提供了一种白平衡计算方法，包括：
[0005]获取初始图像，并根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比；
[0006]若所述第一占比大于预设的第一阈值，计算所述干扰色对应的类白区的第二占比；
[0007]若所述第二占比大于预设的第二阈值，对所述初始图像中的所述干扰色进行抠除处理，得到目标图像；
[0008]对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数
。
[0009]本申请还提供了一种白平衡计算装置，包括：
[0010]第一占比计算单元，用于获取初始图像，并根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比；
[0011]第二占比计算单元，用于若所述第一占比大于预设的第一阈值，计算所述干扰色对应的类白区的第二占比；
[0012]干扰色抠除单元，用于若所述第二占比大于预设的第二阈值，对所述初始图像中的所述干扰色进行抠除处理，得到目标图像；
[0013]白平衡计算单元，用于对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数
。
[0014]本申请还提供一种计算机设备，包括存储器和处理器，所述存储器中存储有计算机程序，所述处理器执行所述计算机程序时实现上述任一项所述的一种白平衡计算方法和
/
或上述任一项所述的白平衡计算方法的步骤
。
[0015]本申请还提供一种计算机可读存储介质，其上存储有计算机程序，所述计算机程序被处理器执行时实现上述任一项所述的一种白平衡计算方法和
/
或上述任一项所述的白
Oxide Semiconductor
，互补金属氧化物半导体
)
或者
CCD(Charge Coupled Device
，电荷耦合器件
)
图像传感器将捕捉到的光源信号转化为数字信号的原始数据
。
在具体的实施方式中，通过图像传感器对需要进行白平衡计算的图像进行采集，得到
RAW
格式的初始图像，再根据预先设定的干扰色，在初始图像中划定干扰色区域，从而计算初始图像中的第一占比
。
其中，由于偏白或偏黄的肤色和绿色往往容易对白平衡结果产生干扰，因此上述干扰色通常可以设置为偏白或偏黄的肤色或绿色
。
[0031]如上步骤
S2
所述，如果第一占比大于提前设置的第一阈值，则可以判定该干扰色对初始图像的整体影响较大，此时需要通过该干扰色对应的类白区的第二占比来判断该干扰色对白区的影响程度
。
其中，上述类白区是指初始图像中靠近白区的区域
。
[0032]如上步骤
S3
所述，在画面中有干扰色的同时，如果第二占比大于预设的第二阈值，即干扰色对应的灰点很多，则可以说明此时的干扰区域不是其他色温的白点区域，而是该干扰色对应的干扰区域，因此需要进行抠除处理
。
在具体的实施方式中，可以直接将抠除的部分从初始图像中剔除，或采用透明涂层对抠除部分进行填充
。
[0033]如上步骤
S4
所述，在干扰色抠除后，对抠除后得到的目标图像进行白平衡计算，具体地，可以根据下式，基于
RGB
三个色彩通道下的灰度均值计算白平衡增益参数
AWB
：
[0034][0035]综上，通过预先设置的干扰色区域对初始图像中的干扰色进行第一占比计算，以及对干扰色对应的类白区进行第二占比计算，从而判定是否需要进行干扰色抠除，避免了干扰色对其他色温白平衡统计的干扰，从而提高白平衡增益参数计算的准确性，避免基于白平衡增益参数增益后的图像出现偏色的情况
。
[0036]在一个实施例中，所述根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比，包括：
[0037]根据
RGB
色彩模式建立色彩坐标系；
[0038]基于所述色彩坐标系，获取所述初始图像在各个色彩通道上的颜色信息；
[0039]根据所述颜色信息识别所述干扰色区域，并计算所述初始图像的干扰色对应的第一占比
。
[0040]如上所述，参照图2，是基于
RGB
色彩模式的，
D65
色温下的
AWB
落点示意图，图中，标号为1的部分为色卡中的类白点，标号为2的部分为肤色块的落点，标号为3的部分为绿色块的落点
。
本实施例采取横坐标为
R/G
，纵坐标为
B/G
的坐标系来统计初始图像中的三个通道的信息
。
[0041]具体地，基于现有技术中的
AWB
算法，如果采取白点统计的方法，当初始图像中存在大面积肤色的情况下，就会影响最终的
AWB
计算结果，而绿色区域的点会被当成中色温的白点被统计出来，使得最终的色温计算结果偏中色温，若直接把肤色和绿色作为干扰色区域直接抠掉，则势必会影响其他色温的白点统计，因此，本实施例中基于白点跟干扰色的占比来决定是否采取抠除的机制
。
[0042]在一个实施例中，所述对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数，包括：
[0043]计算所述目标图像在各个色彩通道中的灰度均值；
[0044]根据所述灰度均值计算第一色彩通道对于第二色彩通道和第三色彩通道的所述白平衡增益参数
。
[0045]如上所述，计算
R(Red)、G(Green)、B(Blue)
三个色彩通道下的灰度均值，可以将
G
通道作为主亮度通道，随后计算
R<本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.
一种白平衡计算方法，其特征在于，包括：获取初始图像，并根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比；若所述第一占比大于预设的第一阈值，计算所述干扰色对应的类白区的第二占比；若所述第二占比大于预设的第二阈值，对所述初始图像中的所述干扰色进行抠除处理，得到目标图像；对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数
。2.
根据权利要求1所述的白平衡计算方法，其特征在于，所述根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比，包括：根据
RGB
色彩模式建立色彩坐标系；基于所述色彩坐标系，获取所述初始图像在各个色彩通道上的颜色信息；根据所述颜色信息识别所述干扰色区域，并计算所述初始图像的干扰色对应的第一占比
。3.
根据权利要求2所述的白平衡计算方法，其特征在于，所述对所述目标图像进行白平衡计算，得到所述目标图像的白平衡增益参数，包括：计算所述目标图像在各个色彩通道中的灰度均值；根据所述灰度均值计算第一色彩通道对于第二色彩通道和第三色彩通道的所述白平衡增益参数
。4.
根据权利要求2所述的白平衡计算方法，其特征在于，所述对所述初始图像中的所述干扰色进行抠除处理，包括：通过所述颜色信息识别所述干扰色区域的边缘；根据所述边缘对所述干扰色进行抠除处理
。5.
根据权利要求1所述的白平衡计算方法，其特征在于，所述根据预先设置的干扰色区域，计算所述初始图像的干扰色对应的第一占比，包括：根据预先设置的干扰色，将所述干扰色所在的区域作为所述干扰色区域；计...

【专利技术属性】
技术研发人员：刘宏波，
申请(专利权)人：广州视源人工智能创新研究院有限公司，
类型：发明
国别省市：