光源值的计算方法、装置、电子设备及介质制造方法及图纸

本申请公开了一种光源值的计算方法、装置、电子设备及介质，属于人工智能技术领域。该光源值的计算方法包括：基于摄像头采集到的图像数据，生成第一图像；基于N个候选光源，提取上述第一图像对应的第一图像特征信息，N为正整数；基于该第一图像特征信息，得到上述N个候选光源对应的预测信息，该预测信息用于表征上述N个候选光源中每个候选光源分别与上述第一图像的真实光源间的相似度；基于上述预测信息和每个上述候选光源的光源值，计算得到预测光源值，该预测光源值用于校正上述第一图像。该预测光源值用于校正上述第一图像。该预测光源值用于校正上述第一图像。

【技术实现步骤摘要】
光源值的计算方法、装置、电子设备及介质


[0001]本申请属于人工智能
，具体涉及一种光源值的计算方法、装置、电子设备及介质。

技术介绍

[0002]随着智能终端技术的发展，电子设备的拍照功能也发生着巨大变化，用户在使用电子设备进行拍照时，更加注重拍摄得到的图像的色彩真实度。
[0003]在相关技术中，电子设备在对目标对象拍照的过程中，需要对摄像头拍摄到的raw格式的图像进行处理，以生成最终的图像。例如，对raw格式的图像中能够反映真实光源值的灰色区域进行处理，以得到照射到该目标对象上的真实光源值的预测值，再基于该预测到的光源值，对图像的原始格式中的非灰色区域进行色彩校正，生成最终的图像。
[0004]然而，采用上述方案，在上述raw格式的图像中不存在灰色区域，或对灰色区域判定错误时，会导致预测到的光源值与照射到该目标对象上的真实光源值出现偏差，从而使得通过该预测光源值校正上述raw格式的图像得到的最终图像，无法表现目标对象的真实色彩。

技术实现思路

[0005]本申请实施例的目的是提供一种光源值的计算方法、装置、电子设备及介质，能够解决预测到的光源值与真实光源值出现偏差的问题。
[0006]第一方面，本申请实施例提供了一种光源值的计算方法，该方法包括：基于摄像头采集到的图像数据，生成第一图像；基于N个候选光源，提取上述第一图像对应的第一图像特征，N为正整数；基于该第一图像特征信息，得到上述N个候选光源对应的预测信息，该预测信息用于表征上述N个候选光源中每个候选光源分别与上述第一图像的真实光源间的相似度；基于上述预测信息和每个上述候选光源的光源值，计算得到预测光源值，该预测光源值用于校正上述第一图像。
[0007]第二方面，本申请实施例提供了一种光源值的计算装置，该装置包括：生成模块、提取模块、预测模块和计算模块，其中：上述生成模块，用于基于摄像头采集到的图像数据，生成第一图像；上述提取模块，用于基于N个候选光源，提取生成模块生成的上述第一图像对应的第一图像特征，N为正整数；上述预测模块，用于基于提取模块提取到的上述第一图像特征信息，得到上述N个候选光源对应的预测信息，该预测信息用于表征上述N个候选光源中每个候选光源分别与上述第一图像的真实光源间的相似度；上述计算模块，用于基于处理模块得到的上述预测信息和每个上述候选光源的光源值，计算得到预测光源值，该预测光源值用于校正上述第一图像。
[0008]第三方面，本申请实施例提供了一种电子设备，该电子设备包括处理器和存储器，所述存储器存储可在所述处理器上运行的程序或指令，所述程序或指令被所述处理器执行时实现如第一方面所述的方法的步骤。
[0009]第四方面，本申请实施例提供了一种可读存储介质，所述可读存储介质上存储程序或指令，所述程序或指令被处理器执行时实现如第一方面所述的方法的步骤。
[0010]第五方面，本申请实施例提供了一种芯片，所述芯片包括处理器和通信接口，所述通信接口和所述处理器耦合，所述处理器用于运行程序或指令，实现如第一方面所述的方法。
[0011]第六方面，本申请实施例提供一种计算机程序产品，该程序产品被存储在存储介质中，该程序产品被至少一个处理器执行以实现如第一方面所述的方法。
[0012]在本申请实施例中，电子设备可以先基于摄像头采集到的图像数据，生成第一图像；再基于N个候选光源，提取该第一图像对应的第一图像特征，N为正整数；接着，基于该第一图像特征信息，得到上述N个候选光源对应的预测信息，该预测信息用于表征上述N个候选光源中每个候选光源分别与上述第一图像的真实光源间的相似度；最后，基于上述预测信息和每个上述候选光源的光源值，计算得到预测光源值，该预测光源值用于校正上述第一图像。如此，由于电子设备可以预测得到每个候选光源与真实光源间的相似度，使得电子设备基于该多个相似度计算得到的预测光源值更为精准，从而，使得通过该预测光源值校正的图像更能表现真实色彩。
附图说明
[0013]图1是本申请实施例提供的一种光源值的计算方法的方法流程示意图之一；
[0014]图2是本申请实施例提供的一种光源值的计算方法所应用的光源值分布图；
[0015]图3是本申请实施例提供的一种光源值的计算方法所应用的函数线性图；
[0016]图4是本申请实施例提供的一种光源值的计算方法的方法流程示意图之二；
[0017]图5是本申请实施例提供的一种光源值的计算装置的结构示意图；
[0018]图6是本申请实施例提供的一种电子设备的结构示意图；
[0019]图7是本申请实施例提供的一种电子设备的硬件示意图。
具体实施方式
[0020]下面将结合本申请实施例中的附图，对本申请实施例中的技术方案进行清楚地描述，显然，所描述的实施例是本申请一部分实施例，而不是全部的实施例。基于本申请中的实施例，本领域普通技术人员获得的所有其他实施例，都属于本申请保护的范围。
[0021]本申请的说明书和权利要求书中的术语“第一”、“第二”等是用于区别类似的对象，而不用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换，以便本申请的实施例能够以除了在这里图示或描述的那些以外的顺序实施，且“第一”、“第二”等所区分的对象通常为一类，并不限定对象的个数，例如第一对象可以是一个，也可以是多个。此外，说明书以及权利要求中“和/或”表示所连接对象的至少其中之一，字符“/”，一般表示前后关联对象是一种“或”的关系。
[0022]下面结合附图，通过具体的实施例及其应用场景对本申请实施例提供的光源值的计算方法、装置、电子设备及介质进行详细地说明。
[0023]在使用相机进行拍照时，相机会自动调度自动白平衡功能，以解决不同相机图像传感器对光源的响应不同而导致的呈现出的照片颜色不同的问题，以及人眼对颜色恒常性
的要求的问题。而颜色恒常性是指，不管是白色光源还是蓝色光源或者黄色光源照在红色的草莓上，人眼看到的草莓都是呈现红色的，但是其实人眼接收的光并不是红色(同理相机接收的光也不是红色)，但是相机所拍摄的照片要还原物体本身的颜色，就需要自动白平衡来调整，使相机拍照得到的照片的颜色和物体的真实颜色一致。
[0024]在相关技术中，电子设备在预测光源值的过程中，所使用的方案大致分为两种：
[0025]一种方案是，采用自动白平衡算法，该算法的一般做法是，基于灰世界假设来进行。而灰世界假设是指，我们所处的世界的颜色的平均值是灰色的(红、绿、蓝三通道的平均值为光源在相机图像传感器上的响应值)。在该假设前提下，电子设备直接对线性图像中的每个像素块取平均值，得到预测光源值R/G和B/G。而采用这种方案，当灰世界假设不满足其假设前提时(如上述线性图像为单一色彩组成的图像)，所预测的光源值与真实的光源值会有很严重的偏差。
[0026]另一种方案是，基于找灰块的算法，该算法是在相机图像传感器的镜头所能覆盖的范围内寻找灰块的算法，由于灰块对光源本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种光源值的计算方法，其特征在于，所述方法包括：基于摄像头采集到的图像数据，生成第一图像；基于N个候选光源，提取所述第一图像对应的第一图像特征信息，N为正整数；基于所述第一图像特征信息，得到所述N个候选光源对应的预测信息，所述预测信息用于表征所述N个候选光源中每个候选光源分别与所述第一图像的真实光源间的相似度；基于所述预测信息和每个所述候选光源的光源值，计算得到预测光源值，所述预测光源值用于校正所述第一图像。2.根据权利要求1所述的方法，其特征在于，在所述第一图像包括第一子图像的情况下，第二图像特征信息为所述第一图像特征信息，所述基于N个候选光源，提取所述第一图像对应的第一图像特征信息，包括：将所述第一子图像输入特征预测模型后，基于所述特征预测模型中的卷积神经网络，对所述第一子图像进行特征提取，得到每个所述候选光源下的所述第一子图像的语义特征信息；基于所述特征预测模型中的全局池化层，对所述语义特征信息进行特征转换，得到每个所述候选光源下的所述第一子图像的第二图像特征信息，其中，所述第二图像特征信息为所述第一图像特征信息。3.根据权利要求1所述的方法，其特征在于，在所述第一图像包括第一子图像和第二子图像的情况下，所述基于N个候选光源，提取所述第一图像对应的第一图像特征信息，包括：将所述第一子图像输入特征预测模型后，基于所述特征预测模型中的卷积神经网络，对所述第一子图像进行特征提取，得到每个所述候选光源下的所述第一子图像的语义特征信息；基于所述特征预测模型中的全局池化层，对所述语义特征信息进行特征转换，得到每个所述候选光源下的所述第一子图像的第二图像特征信息；将所述第二图像特征信息与所述第二子图像中包含的色温特征信息以及灰度世界假设特征信息拼接，得到所述第一图像特征信息。4.根据权利要求1至3任一项所述的方法，其特征在于，所述基于所述第一图像特征信息，得到所述N个候选光源对应的预测信息，包括：将所述第一图像特征信息输入特征信息分类模型，对所述第一图像特征信息进行分类，得到所述N个候选光源对应的预测信息。5.根据权利要求1所述的方法，其特征在于，所述基于所述预测信息和每个所述候选光源的光源值，计算得到预测光源值之后，所述方法还包括：基于所述预测信息和所述N个候选光源的光源值，计算所述N个候选光源的离散程度值；基于所述预测光源值和所述真实光源的光源值，计算所述预测光源值与所述真实光源的光源值的角度误差值；将所述离散程度值和所述角度误差值进行拟合，得到拟合函数，所述拟合函数用于表征候选光源的离散程度与角度误差间的关联关系；将所述预测光源值与所述真实光源的光源值间的离散程度值，输入所述拟合函数，得到目标角度误差值；
其中，所述目标角度误差值用于指示所述预测光源值是否有效。6.一种光源值的计算装置，其特征在于，所述装置包括：生成模块、提取模块、预测模块和计算模块，其中：所述生成模块，用于基于摄像头采集到的图像数据，生成第一图像；所述提取模块，用于基...

【专利技术属性】
技术研发人员：何月华，
申请(专利权)人：维沃移动通信有限公司，
类型：发明
国别省市：