基于改进SqueezeNet的颜色恒常性计算  

Color Constancy Calculation Based on Improved SqueezeNet

作　　者：张金友 张霞 徐亮山 ZHANG Jin-You;ZHANG Xia;XU Liang-Shan(Research Center of Graphic Communication,Printing and Packaging,Wuhan University,Wuhan 430079,China;School of Civil Engineering,Wuhan University,Wuhan 430079,China;Shenzhen Tencent Computer System Co.Ltd.,Shenzhen 510310,China)

机构地区：[1]武汉大学图像传播与印刷包装研究中心,武汉430079 [2]武汉大学土木建筑工程学院,武汉430079 [3]深圳市腾讯计算机系统有限公司,深圳510310

出　　处：《计算机系统应用》2023年第10期208-214,共7页Computer Systems & Applications

基　　金：国家自然科学基金(62175191)。

摘　　要：由于成像设备存在的缺陷,容易引起成像色彩的偏移,影响图像算法的下游任务,因此需要采用颜色恒常性算法实现图像色彩的矫正,保证图像颜色与人眼看到的色彩保持一致.传统颜色恒常性算法的效果依赖于特定的光源环境,为了提升算法的适用范围和使用效率,提出了一种基于SqueezeNet框架的颜色恒常性计算模型,通过卷积图像网络感知图像光源,并引入了注意力机制和残差连接,提升网络对图像的理解和计算性能.网络同时预测输入图像各区域的光照颜色,再通过设计3种不同池化方式汇聚,输出图像的全局估计光源,最后利用估计光源矫正图像.实验结果表明,提出的光源估计算法能够有效地估计图像光照颜色,矫正图像色彩.Due to the defects of imaging equipment,it is easy to cause the shift of the imaging color and affect the downstream tasks of the image algorithm.Therefore,the color constancy algorithm is required to correct the image color so that the image color is consistent with the color seen by the human naked eye.The effect of the traditional color constancy algorithm depends on specific light source environments.In order to improve the application range and utilization efficiency of the color constancy algorithm,a color constancy calculation model based on the SqueezeNet framework is proposed,which senses the image light source through the convolutional image network.In addition,the attention mechanism and residual connection are introduced to improve the network’s understanding of images and computing performance.The network predicts the illumination color of each area of the input image at the same time and then gathers them by designing three different pooling methods to output the global estimated light source of the image,and it finally uses the estimated light source to correct the image.Experimental results show that the proposed light source estimation algorithm can effectively estimate the illumination color of the image and correct the image color.

关 键 词：颜色恒常性 光源估计 SqueezeNet 注意力机制 残差连接 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]