[发明专利]基于约束粒子群算法的色彩校正方法、装置及终端设备

说明书

本发明适用于图像处理技术领域，提供了一种基于约束粒子群算法的色彩校正方法、装置及终端设备，方法包括：基于待校正的目标图片获取校正函数；通过随机排序的粒子群算法计算校正函数的最优解；其中，粒子群算法的迭代次数小于预设迭代次数时，以第一条件接收不满足约束条件的校正函数最优解，粒子群算法的迭代次数大于预设迭代次数时，以第二条件接收不满足约束条件的校正函数最优解，第一条件要求的范围大于第二条件要求的范围；根据具有最优解的校正函数校正目标图片的色彩。通过本发明可以根据具有最优解的校正函数得到最佳的色彩校正矩阵，准确还原目标图片的色彩，提高色彩校正效果。

技术领域

本发明涉及图像处理技术领域，尤其涉及一种基于约束粒子群算法的色彩校正方法、装置及终端设备。

背景技术

色彩校正通过3乘3的色彩校正矩阵对原始的RGB值做线性变换，逼近真实的RGB值，为了使得校正后的色彩尽可能的逼近真实的色彩，需要计算到最优色彩校正矩阵。同时，还必须满足矩阵的每一行之和等于1的约束条件，这样能够确保色彩校正后白色物体(R，G，B值相等或者近似相等)依然是白色的，且亮度不改变。

然而，目前使用的色彩校正方法，如最小二乘法、遗传算法等，通常都无法保证校正矩阵每一行之和为1的要求，使得色彩校正过早的陷入局部最优或者不满足约束条件的情况发生，导致色彩校正结果不精准，校正后的图片色彩还原程度不高。

发明内容

本发明的主要目的在于提出一种基于约束粒子群算法的色彩校正方法、装置及终端设备，以解决现有技术中色彩校正方法的色彩校正结果不精准，校正后的图片色彩还原程度不高的问题。

为实现上述目的，本发明实施例第一方面提供一种基于约束粒子群算法的色彩校正方法，包括：

基于待校正的目标图片获取校正函数；

通过随机排序的粒子群算法计算所述校正函数的最优解；

其中，所述粒子群算法的迭代次数小于预设迭代次数时，以第一条件接收不满足约束条件的校正函数最优解，所述粒子群算法的迭代次数大于预设迭代次数时，以第二条件接收不满足约束条件的校正函数最优解，所述第一条件要求的范围大于所述第二条件要求的范围；

根据具有最优解的校正函数校正所述目标图片的色彩。

结合本发明第一方面，本发明第一实施方式中，通过随机排序的粒子群算法计算所述校正函数的最优解，包括：

获取所述校正函数中的校正矩阵A_m，其中，m为迭代次数；

初始化所述校正矩阵A_m的N个元素，其中，每个元素包括速度属性、位置属性、适应度属性和约束背离属性；

所述迭代次数m＝1时校正矩阵A₁为初始校正矩阵，根据所述初始校正矩阵直接计算所述校正函数的最优解；

所述迭代次数m大于1时，计算校正矩阵A_CurGen中的个体元素最优解的位置、适应度、约束背离值，全局元素最优解、所述全局元素最优解的位置、适应度、约束条件背离值；

根据所述校正矩阵A_CurGen中的个体元素最优解和全局元素最优解的位置更新所述校正矩阵A_CurGen中每个元素的速度和位置得到新的矩阵A_CurGen+1，，并返回初始化所述校正矩阵A_CurGen+1，的N个元素，直至所述迭代次数达到总的迭代次数，或所述校正矩阵A_CurGen+1，中的全局元素最优解没有更新为止；

使用最终的校正矩阵计算所述校正函数的最优解；

其中，CurGen为当前迭代次数，CurGen小于或等于m，m＝1,2,3...,MAXGEN；MAXGEN为总的迭代次数，N为正整数；