[发明专利]源图像数据库的获取方法、训练装置及电子设备

说明书

本发明涉及一种源图像数据库的获取方法及获取系统、神经网络的训练方法、训练装置、计算机可读存储介质、图像处理芯片、图像处理系统、电子设备，获取方法包括：首先分别获取已知图像库和校正图像库，并分别计算各自图像对应的光源信息；然后根据多组色温点相同的原始图像和采集图像的光源信息，建立色温模型，并根据色温模型对已知图像库的各原始图像的光源信息进行转换，得到目标图像库中各目标图像的光源信息；接着根据已知图像库和目标图像库的光源信息将已知图像库的原始图像的像素信息转换到目标图像库的目标图像的像素信息，得到目标采集单元对应的源图像数据库。本发明极大地减少了神经网络训练之前的图像采集数量。

技术领域

本发明涉及图像处理领域，具体涉及一种源图像数据库的获取方法及获取系统、神经网络的训练方法、训练装置、计算机可读存储介质、图像处理芯片、图像处理系统、电子设备。

背景技术

白平衡指在任何光源下都可以将白色物体还原为白色的方法。人眼具有适应性，所以我们有时候不能发现色温的变化，但是电子设备并不具备这样的特点，所以需要白平衡算法来还原色彩。现有技术中，许多白平衡处理用的都是传统算法，如灰度世界法，完美反射法等，这些算法的核心思想都是直接找到图像中的白块进行色彩还原，但是，真实世界的场景是复杂多变的，一旦场景中白块较少或没有白块，那么算法就会失效，这也导致其鲁棒性不强。一种较先进的白平衡处理算法是采用神经网络进行，具体地，目标采集单元采集大量的raw图像，然后对神经网络进行训练，得到训练好的模型，采用训练好的模型对目标采集单元采集的图像进行白平衡处理。

上述这种神经网络的方法虽然性能较好，但是其必须依靠大量的raw图像数据来支撑，而对于不同镜头或者传感器对应的色温曲线也不尽相同，这就意味着不同的电子设备，只要镜头和传感器有一者发生变化，就需要重新采集大量的raw图像对神经网络进行重新训练，对于设备厂商来说，会制造多种设备，各设备的镜头和传感器常常会更换，即使是同一型号的设备，也常常会发生变化，这就造成每台设备都需要重新采集大量的raw图像，导致神经网络之前的采集工作急剧增加，提高了设备的生产时间及成本。

发明内容

基于上述现状，本发明的主要目的在于提供一种源图像数据库的获取方法及获取系统、神经网络的训练方法、训练装置、计算机可读存储介质、图像处理芯片、图像处理系统、电子设备，以解决由于镜头和传感器的更换导致的工作量大大增加的问题。

为实现上述目的，本发明采用的技术方案如下：

本发明的第一方面提供了一种源图像数据库的获取方法，所述源图像数据库用于训练神经网络，所述神经网络用于对采集单元采集的图像进行白平衡处理，所述获取方法包括步骤：

S100：获取已知图像库，并得到所述已知图像库中各原始图像对应的光源信息；其中，所述已知图像库中的所述原始图像覆盖多个色温点，且各所述原始图像中带有标准色卡；

S200：获取校正图像库，并计算所述校正图像库中各采集图像对应的光源信息；其中，所述校正图像库通过使用目标采集单元采集不同色温下的带标准色卡的采集图像形成；

S300：根据多组色温点相同的所述原始图像和所述采集图像中对应的光源信息，建立由所述原始图像转换到所述采集图像的色温模型；所述色温模型包括转换矩阵c和转换系数z₀，其中，所述原始图像的光源信息与所述转换矩阵c的乘积等于所述采集图像的光源信息与转换系数的乘积；

S400：根据所述色温模型，对所述已知图像库的各所述原始图像的光源信息进行转换，得到目标图像库中各目标图像的光源信息；

S500：根据所述已知图像库和所述目标图像库的光源信息将所述已知图像库的各所述原始图像的像素信息转换到所述目标图像库的各所述目标图像的像素信息，得到所述目标采集单元对应的源图像数据库；