[发明专利]一种白平衡处理方法及相关装置

说明书

本申请公开了一种白平衡处理方法及相关装置，该方法包括：对待处理图像进行目标分割，得到目标分割区域；确定目标分割区域的属性信息；根据目标分割区域的属性信息确定待处理图像的图像类型，其中，图像类型包括：单色图像、普通场景图像和色彩丰富图像；利用与待处理图像的图像类型匹配的白平衡校正方式，对待处理图像进行白平衡处理。本申请所提供的技术方案，可实现根据图像的图像类型对待处理图像做出针对性的白平衡校正。

技术领域

本申请涉及图像处理领域，特别是涉及一种白平衡处理方法及相关装置。

背景技术

人类视觉系统具有对物体真实色彩的还原能力，无论是在室外自然光下，还是在室内荧光灯或者混合色温光源下，人眼都会将白色物体识别为白色。由于摄像机对白色的响应与人眼不同，导致白色物体在不同的环境光源下所呈现出的颜色存在差异。因此需要对摄像机采集到的图像进行白平衡校正，把采集到的图像中白色物体修正为白色。在现有的白平衡处理方案中，无法根据图像场景做出针对性校正，故需要一种可以解决上述技术问题的技术方案。

发明内容

本申请主要解决的技术问题是提供白平衡处理方法及相关装置，可以实现根据图像的图像类型对待处理图像做出针对性的白平衡校正。

为解决上述技术问题，本申请采用的一个技术方案是：提供一种白平衡处理方法，所述方法包括：

对待处理图像进行目标分割，得到目标分割区域；

确定所述目标分割区域的属性信息；

根据所述目标分割区域的属性信息确定所述待处理图像的图像类型，其中，所述图像类型包括：单色图像、普通场景图像和色彩丰富图像；

利用与所述待处理图像的图像类型匹配的白平衡校正方式，对所述待处理图像进行白平衡处理。

进一步地，所述对待处理图像进行目标分割，得到目标分割区域，包括：

利用图像分割模型对所述待处理图像进行目标分割，得到若干初始分割区域；

统计每个所述初始分割区域中的灰度值的离散程度；

分别判断每个所述初始分割区域对应的离散程度是否大于或等于第一预设阈值；

根据判断结果对所述初始分割区域进行二次分割，以获得所述目标分割区域。

更进一步地，所述根据判断结果对所述初始分割区域进行二次分割，以获得所述目标分割区域，进一步包括：

若所述初始分割区域的离散程度大于或等于所述第一预设阈值，则根据所述初始分割区域的颜色信息对所述初始分割区域进行二次分割，以获得所述目标分割区域；或

若所述初始分割区域的离散程度小于所述第一预设阈值，则直接将所述初始分割区域作为所述目标分割区域。

进一步地，所述目标分割区域的属性信息至少包括：所述待处理图像中包括的所述目标分割区域的区域数量、最大尺寸的目标分割区域在所述待处理图像中的占比。

更进一步地，确定所述目标分割区域的属性信息，进一步包括：

统计所述待处理图像中包括的所述目标分割区域的区域数量、每个目标分割区域所包括的像素点数量和所述待处理图像所包括的像素点数量，并确定所述像素点数量最多的目标分割区域为所述最大尺寸的目标分割区域；

利用所述最大尺寸的目标分割区域的像素点数量与所述待处理图像的像素点数量，得到所述最大尺寸的目标分割区域在所述待处理图像中的占比。

更进一步地，所述根据所述目标分割区域的属性信息确定所述待处理图像的图像类型，进一步包括：

若所述区域数量小于或等于第二预设阈值，则确定所述待处理图像的图像类型为所述单色图像；