一种融合光源颜色先验信息的NetQuery网络多光源颜色估计方法

摘要

本发明公开了一种融合光源颜色先验信息的NetQuery网络多光源颜色估计方法，具体包括两个阶段，第一阶段是训练阶段，对原始图像进行处理得到光源先验信息，并建立训练数据集，然后构建NetQuery网络，使用训练数据集对网络进行训练；第二阶段是使用阶段，输入原始图像，通过多光源颜色恒常性方法得到该原始图像的光源颜色先验信息，将使用原始图像及其对应的光源颜色先验信息输入训练好的NetQuery网络，得到估计的多光源分布。本发明的方法将光源颜色先验信息融入进深度学习模型，能进一步提升深度学习网络对多光源图像的校正效果；通过有效的将输入图像与光源先验信息进行编码，并融合编码特征，进行精细的多光源估计，能够很好的实现多光源颜色恒常性。