一种用于红外耦合图像色彩校正的生成对抗网络

CCD/CMOS传感器在图像采集过程中，R,G,B三通道都会受到红外光线的串扰，导致所拍摄的照片呈现一种不正常的红色状态。针对这一问题，文中提出一种用于红外耦合图像色彩校正的生成对抗网络。生成网络使用编码解码结构。编码器通过卷积层提取红外耦合图像特征；解码器对提取到的特征进行逆向求解、重构，以剔除图像红、绿、蓝通道中混叠的红外信号。为了进一步缩小生成图像与源图像的差距，使其在颜色、纹理细节等方面更具一致性，文中在原有的对抗损失基础上加入了MSE损失优化网络参数，对网络模型进行约束。实验结果表明，相较于其他色彩校正算法，使用文中方法校正的图像色彩更加鲜明，纹理细节更加突出，更加符合人眼的直观感受；PSNR,SSIM,AVG指标分别比Gray World算法高8.447 1 dB,0.082 3,1.347 1,CD,NIQE指标分别比Gray World算法低6.176 8,1.064 9,具有更好的校正效果。