将场景光照建模为风格以实现自动白平衡校正

风格可指代不同概念（如绘画风格、发型、纹理、颜色、滤镜等），具体取决于特征空间如何形成。在本工作中，我们提出一种新颖思路，将单光源与多光源场景中的光照解释为一种风格概念。为验证该思路，我们引入一种增强的自动白平衡（AWB）方法，将单光源与混合光源场景中的光照建模为风格因子。我们的 AWB 方法不需要任何光照估计步骤，但包含一个网络，用于生成具有不同白平衡（WB）设置的图像的权重图。所提网络利用了由多头风格提取模块从场景中提取的风格信息。在将这些权重图与场景融合后，即完成 AWB 校正。在单光源与混合光源数据集上的实验表明，与近期工作相比，我们所提方法取得了有前景的校正结果。这表明具有多光源场景中的光照可由风格概念来建模。源代码与训练模型可在 https://github.com/birdortyedi/lighting-as-style-awb-correction 获取。