[发明专利]图像处理方法、装置、存储介质以及终端

说明书

本申请实施例公开了一种图像处理方法、装置、存储介质以及终端，所述方法包括：获取相机采集的目标图像的颜色通道数据，将颜色通道数据输入至光源分类模型中，得到目标图像对应的光源属于不同预设类型光源的匹配概率，基于匹配概率确定目标图像对应的白平衡增益，以及基于白平衡增益对目标图像进行白平衡调整。本申请通过光源分类模型对目标图像的颜色通道数据进行识别，颜色通道数据实际上为不同波段的光谱通道数据，也就是通过光源分类模型对采集的不同波段的光谱通道数据进行识别，可以准确地得到目标图像对应的光源属于不同预设类型光源的匹配概率，再依据匹配概率得到目标图像的白平衡增益，可以提高对图像进行白平衡的准确性。

技术领域

本申请涉及图像处理技术领域，尤其涉及一种图像处理方法、装置、存储介质以及终端。

背景技术

白平衡(Auto White Balance，简称AWB)是指在不同的色温下，使传感器采集的图像接近人眼所看的真实画面。也就是说，AWB可以使白色的物体在不同的色温下拍摄得到的图像也是白色的。在日常使用中，手机的相机可以设定自动白平衡，自动白平衡在大部分情况下就可以满足用户拍摄的需求。利用自动白平衡技术，相机能够自动估计环境色光，并去除光源对图像传感器的影响，使相机能够具有与人类视觉系统一样的颜色恒常性。

发明内容

本申请实施例提供了一种图像处理方法、装置、计算机存储介质以及终端，可以提高对图像进行白平衡处理的准确性。所述技术方案如下：

第一方面，本申请实施例提供了一种图像处理方法，所述方法包括：

获取相机采集的目标图像的颜色通道数据；

将所述颜色通道数据输入至光源分类模型中，得到所述目标图像对应的光源属于不同预设类型光源的匹配概率；

基于所述匹配概率确定所述目标图像对应的白平衡增益，以及基于所述白平衡增益对所述目标图像进行白平衡调整。

第二方面，本申请实施例提供了一种图像处理装置，所述装置包括：

数据采集模块，用于获取相机采集的目标图像的颜色通道数据；

光源识别模块，用于将所述颜色通道数据输入至光源分类模型中，得到所述目标图像对应的光源属于不同预设类型光源的匹配概率；

图像调整模块，用于基于所述匹配概率确定所述目标图像对应的白平衡增益，以及基于所述白平衡增益对所述目标图像进行白平衡调整。

第三方面，本申请实施例提供了一种计算机存储介质，所述计算机存储介质有多条指令，所述指令适于由处理器加载并执行上述的方法步骤。

第四方面，本申请实施例提供了一种终端，可包括：存储器和处理器；其中，所述存储器存储有计算机程序，所述计算机程序适于由所述存储器加载并执行上述的方法步骤。

本申请实施例提供的技术方案带来的有益效果至少包括：

本申请实施例的方案在执行时，获取相机采集的目标图像的颜色通道数据，将所述颜色通道数据输入至光源分类模型中，得到所述目标图像对应的光源属于不同预设类型光源的匹配概率，基于所述匹配概率确定所述目标图像对应的白平衡增益，以及基于所述白平衡增益对所述目标图像进行白平衡调整。本申请通过光源分类模型对目标图像的颜色通道数据进行识别，颜色通道数据实际上为不同波段的光谱通道数据，也就是通过光源分类模型对采集的不同波段的光谱通道数据进行识别，可以准确地得到目标图像对应的光源属于不同预设类型光源的匹配概率，再依据匹配概率得到目标图像的白平衡增益，可以提高对图像进行白平衡处理的准确性。

附图说明