基于神经网络的颜色恒常性—Fully Convolutional Color Constancy with Confidence-weighted Pooling...

 论文地址：

http://openaccess.thecvf.com/content_cvpr_2017/papers/Hu_FC4_Fully_Convolutional_CVPR_2017_paper.pdf

源代码（Python）:

https://github.com/yuanming-hu/fc4

 

一、    任务描述

网络的主要目的是能够对偏色的图片估计光源，从而移除偏色，恢复图片真实颜色。为满足此类网络训练要求，需要数据集中不仅包括图片且需要提供图片的真实光源数据。

 

二、    数据集

解释网络原理前，先下载数据集并了解数据集，有助于后面原理的理解

数据集（Shi's Re-processing of Gehler's Raw Dataset）：

http://www.cs.sfu.ca/~colour/data/shi_gehler/

下载的文件包括:

png_canon1d.zip

png_canon5d_1.zip

png_canon5d_2.zip

png_canon5d_3.zip

groundtruth_568.zip；

其中，前四个文件夹中是图片，最后一个文件夹中包括对应图片的光照数据。

下载后，将所有图片放入一个文件夹中，一共有568张图片

注：568张图片是16位RAW图像，电脑图片查看工具一般是8位的，所以显示纯黑

 

利用Python代码进行图片格式转换(训练时中并不需要此步骤)，可以看到图片真实效果（借助下载文件中保