苏州沐光技术有限公司

本发明公开了一种多光源自动白平衡方法，具体包括以下步骤：获取待处理的输入图像；对输入图像进行特征提取，将输入图像输入编码器‑解码器网络，提取包含全局语义与局部纹理的多尺度图像特征，获得瓶颈特征；进行全局色度基估计，对瓶颈特征进行全局池化，并通过全连接层输出K个全局色度基；进行空间权重估计，由编码器‑解码器网络的解码端输出与K个全局色度基一一对应的K张空间权重响应图，对K张空间权重响应图在基维度执行归一化；生成照明色度图，将每个全局色度基与其对应的空间权重图逐像素相乘后求和，生成二维空间照明色度图；白平衡校正，对输入图像执行逐像素颜色校正，输出白平衡校正后的图像。