摘 要: | 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,具有更好的校正效果。
|