一种使用神经网络进行图像白平衡校正的方法及装置
摘要:
一种使用神经网络进行图像白平衡校正的方法及装置,属于数字图像处理技术领域。包括:S1、采集原始图像;S2、环境光源色温的参数值计算;S3、对所述原始图像和所述待处理图像分别进行预处理得到训练用的预处理图像和第二预处理图像;S4、构建卷积神经网络模型;S5、训练卷积神经网络模型;S6、利用卷积神经网络计算得到红色通道增益值gainR及蓝色通道增益值gainB;S7、利用所述红色通道增益值gainR及蓝色通道增益值gainB对待处理图像进行白平衡校正,从而得到校正后的白平衡图像。该方法解决了现有技术中白平衡容易失效的技术问题,有效的提高了算法的计算速度,精确度得到很大提高,且该模型鲁棒性非常好。
0/0