郑州大学学报(理学版)  2019, Vol. 51 Issue (2): 29-33  DOI: 10.13705/j.issn.1671-6841.2018180

引用本文  

孔韦韦, 雷阳, 任聪, 等. 基于改进型CNN的多聚焦图像融合方法[J]. 郑州大学学报(理学版), 2019, 51(2): 29-33.
KONG Weiwei, LEI Yang, REN Cong, et al. Method for Multi-focus Image Fusion Based on Improved Convolutional Neural Network[J]. Journal of Zhengzhou University(Natural Science Edition), 2019, 51(2): 29-33.

基金项目

国家自然科学基金项目(61309008,61309022);信息保障技术重点实验室开放基金项目(KJ-17-105);陕西省自然科学基础研究计划面上项目(2018JM6047)

通信作者

雷阳(1984—),女,陕西西安人,讲师,主要从事图像智能信息处理研究,E-mail: surina526@163.com

作者简介

孔韦韦(1983—),男,安徽合肥人,副教授,主要从事图像智能信息处理研究,E-mail:kwwking@163.com

文章历史

收稿日期:2018-06-13
基于改进型CNN的多聚焦图像融合方法
孔韦韦1 , 雷阳2 , 任聪2 , 赵文阳1     
1. 西安邮电大学 计算机学院  陕西 西安 710121;
2. 武警工程大学 密码工程学院  陕西 西安 710086
摘要:针对多聚焦图像融合问题,提出一种基于改进型卷积神经网络(convolutional neural network, CNN)的图像融合方法.首先,阐述了多聚焦图像融合问题与CNN中分类问题的一致性关系;其次,对经典的CNN模型加以改进,构建了改进型CNN模型;最后,将待融合源图像输入改进型CNN模型进行实验仿真.仿真结果表明,与现有的代表性融合方法相比,该方法无论在直观视觉效果还是客观评价指标方面均具有显著的优势.
关键词图像融合    多聚焦    卷积神经网络    池化层    
Method for Multi-focus Image Fusion Based on Improved Convolutional Neural Network
KONG Weiwei1 , LEI Yang2 , REN Cong2 , ZHAO Wenyang1     
1. School of Computer Science, Xi′an University of Posts and Telecommunications, Xi′an 710121, China;
2. School of Crytographic Engineering, Engineering University of Armed Police Force, Xi′an 710086, China
Abstract: A method for multi-focus image fusion based on improved convolutional neural network (CNN), was proposed to deal with the issue of multi-focus image fusion. First of all, the consistent relation between multi-focus image fusion and the classification issue in CNN was elaborated. Then, the traditional CNN model was modified to be an improved CNN model. Finally, the source images were input into the improved CNN model to conduct the simulation experiments. Experimental results indicated that, compared with the current typical fusion methods, the proposed method had remarked superiorities in terms of subjective visual performance and objective evaluation results.
Key words: image fusion    multi-focus    convolutional neural network    pooling layer    
0 引言

目前,主流的多聚焦图像融合方法可分为变换域融合方法与空间域融合方法两类.代表性变换域融合方法主要基于多尺度几何分析理论展开,可分为源图像多尺度分解、子图像融合以及最终图像重构3个步骤.经典的小波变换仅能捕获有限方向的图像特征,且不具备平移不变性,非下采样轮廓波变换[1-2]与非下采样剪切波变换[3-7]作为经典轮廓波变换与剪切波变换的改进型模型,极大地提升了图像的融合效果.然而,在此过程中,信息的损失和光谱信息的扭曲仍然是无法回避的问题.与多尺度几何分析方法不同,稀疏表达[8-10]模型侧重于将源图像转换为单一尺度的特征映射,并采用滑动窗口实现平移不变性图像融合的近似表达,该模型具有较好的图像融合效果,但在源图像高频细节信息保存方面略显不足.相比变换域融合方法,早期经典的空间域融合方法通常采取将源图像划分为若干块,而后选取空间频率或梯度信息等评价指标分别对源图像的对应块加以比较和融合.但是图像块的尺寸以及融合规则的选取通常会对最终融合效果造成很大的影响.因此,目前主流的空间域方法主要着眼于图像块尺寸的自适应调整以及融合规则的设定.

CNN作为深度学习中的代表性模型,由于具有优越的图像特征捕捉和信息分析能力,被广泛应用于图像识别、图像检索等领域.本文采用CNN模型对多聚焦图像融合问题加以解决.针对多聚焦图像融合问题与CNN中分类问题的一致性,探索二者的切合点以及本文方案的可行性;对经典的CNN模型加以优化,构建改进型CNN模型;将待融合源图像输入到改进型CNN模型进行实验仿真.仿真结果表明,与现有的代表性融合方法相比,该方法无论在直观视觉效果还是客观评价指标值均具有显著的优势.

1 一致性关系研究

CNN是一个可训练的多层前馈人工神经网络,且每一层都包含与抽象层相对应的一定数量的特征映射.其中,特征映射中的每个单元或系数称为神经元.线性卷积、非线性激活和空间域池化用于实现不同阶段特征映射的连接.图 1给出了经典CNN模型结构图.

图 1 经典CNN模型结构图 Fig. 1 Structure of the traditional CNN model

图 1中的CNN模型拥有1个输入层、1个输出层、3个卷积层以及3个池化层.输入层/输出层用于接收/输出图像信息;CNN的第一卷积层的滤波器主要用于检测低阶特征,比如边、角、曲线等.随着卷积层的增加,对应滤波器检测的特征也更复杂.例如,第二卷积层的输入实际上是第一层的输出,该层滤波器用来检测低阶特征的组合等情况(半圆、四边形等),如此累积,以检测越来越复杂的特征.池化层周期性地出现在相邻的两个卷积层之间,可以有效减少训练参数的数量,压缩图像的空间大小.

CNN模型中滤波器激活图源于特征提取,而融合规则的作用恰恰类似于分类任务中使用的分类器.就现行绝大多数空间域和变换域方法而言,激活映射图可以通过构造局部滤波器来获取源图像的高频细节信息,而融合规则实质上完成了针对源图像像素点的权值分配.CNN模型中的“卷积”操作与多聚焦图像融合过程中的滤波器操作等价,而多聚焦图像的焦点映射图生成过程等价于CNN模型的分类过程.因此,可以采用CNN模型用于图像融合问题的解决.

2 改进型CNN模型

针对经典CNN模型中权值设置随意,以及部分结构不合理的问题,提出并构建一种改进型CNN模型,该模型主要涉及权值设定、卷积层和池化层3个部分.

2.1 权值设定

经典CNN模型涉及大量权值的设定,这些权值设置通常较为随意,而权值设定的优劣直接影响CNN模型的训练效果以及最终的分类结果.针对该问题,本文提出采用高斯概率分布函数确定输入层与第一卷积层(C1)、池化层与相邻后续卷积层间的权值,

$ \mathit{\boldsymbol{P}} \in {{\bf{R}}^{\left( {r \times r} \right) \times N}} = \left[ {{{\hat P}^1},{{\hat P}^2}, \cdots ,{{\hat P}^i}, \cdots ,{{\hat P}^N}} \right],1 \le i \le N, $

其中:矩阵P中每一元素的尺寸均为r×rN表示卷积核的数量.显然,若前一层图像信息的尺寸为k×k,则通过高斯概率分布函数获得的特征映射图的尺寸为(k-r+1)×(k-r+1).

2.2 卷积层

卷积层主要针对源图像或前一池化层中的特征映射采取“卷积”操作,以提取图像中的特征信息.有了2.1节的权值设定,可以得到第i幅特征映射中位于点(x, y)的卷积结果为

$ {c_{x,y,i}}\left( \mathit{\Theta } \right) = \sum\limits_{m = 1}^r {\sum\limits_{n = 1}^r {{\mathit{\Theta }_{x + m - 1,y + n - 1}} \cdot p_{m,n}^i} } , $

其中:Θ表示待融合源图像.需要说明的是,不同于经典CNN模型,特征映射并不适用于非线性函数.

2.3 池化层

池化层通常周期性地出现在相邻的两个卷积层之间,其功能主要在于有效减少训练参数的数量,从而压缩图像的空间大小.经典CNN模型采用平均值池化层策略和最大值池化层策略.然而,这两种策略均有不同程度的缺陷.其中,平均值池化针对每个池化邻域内的结点进行算术平均运算,但会削弱强激活结点产生弱池化效应;最大值池化倾向于过度拟合训练集从而影响泛化性能.为了克服这两种经典池化策略的消极影响,本文提出了均方根池化策略.

大量仿真实验结果表明,即使随机设定若干权值,均方根池化策略仍然具有固有的频率选择性和平移不变性.该池化策略具有平方运算和求和运算两个部分,可以使整个CNN模型具有很好的非线性和平移不变性.将均方根池化策略应用于所有池化层,可以得到点(x, y)位于第j幅池化映射中的结果为

$ {s_{p, q, j}}\left( \mathit{\Theta } \right) = \sqrt {\sum\limits_{x = \left( {p - e} \right)}^{p + e} {\sum\limits_{y = \left( {q - e} \right)}^{q + e} {{\mathit{\Theta }^2}_{x, y, j}} } } , $

其中:Θ表示待融合源图像;p, q=1, …, (k-r+1);e为池化邻域半径.

3 实验仿真

为了验证文中提出的融合方法的有效性,本节将引入几种近年代表性的方法与之进行比较.本文的实验平台为一台CPU,主频为2.9 GHz、8G内存的PC,采用的仿真软件为Matlab 2014b.本节选取了两组灰度图像作为源图像进行融合仿真实验.

3.1 融合方法及量化评价指标

本文将采用性能较为优越的4种融合方法与文中方法(记为M5)进行融合效果比较:基于框架变换耦合SUSAN的多聚焦图像融合算法[11](M1)、基于小波变换的多聚焦图像融合方法[12](M2)、基于NSCT耦合区域特性的多聚焦图像融合算法[13](M3)、基于多尺度CNN和图像分割的多聚焦图像融合算法[14](M4).其中M1~M4中的参数均按照原对应文献中的给定数据设定.本文所提方法的改进型CNN模型具有1个输入层、1个输出层、3个卷积层(3个卷积层的滤波器数量分别为5、10、15,且所有卷积核的尺寸均为3×3),3个池化层(池化层最后一层的邻域尺寸设定为2×2,其余池化层的邻域尺寸设定为5×5).

除了直观视觉效果外,本文还采用了4种客观评价指标比较5种方法的性能,分别为空间频率(spatial frequency, QSF)、Piella指标(QPiella)、互信息量(mutual information, QMI)和相位一致性(phase congruency, QP).QSF是衡量图像像素点信息反差度的重要指标,融合图像的QSF值越大,表明融合图像的信息量越丰富,融合质量越好;QPiella描述了图像的结构信息,QPiella值越大,图像的结构信息越丰富,效果越好;QMI用来衡量融合图像与源图像间的交互信息,QMI值越大,表示融合图像从源图像中获取的信息越丰富,融合效果越好,本文使用的是归一化互信息量;QP利用相位一致性描述图像边缘信息,QP值越大,表明融合图像的边缘信息越丰富.以上各指标的表达式参见文献[15].

3.2 多聚焦图像融合实验

本节选取两组灰度图像作为待融合源图像,图像像素大小均为512×512.该两组图像分别拥有不同的聚焦信息,如何尽可能地将这些聚焦信息融合到一幅结果图像中是本节实验的主要目的.M1~M5 5种方法的图像融合仿真效果如图 2~3所示.

图 2 第Ⅰ组多聚焦图像融合效果图 Fig. 2 Fused results of the multi-focus images in group Ⅰ

图 3 第Ⅱ组多聚焦图像融合效果图 Fig. 3 Fused results of the multi-focus images in group Ⅱ

从直观角度看,上述5种方法均较好地保持了两幅源图像的重要信息,并对源多聚焦图像进行了较好的融合.为了更方便进行比较,本节针对融合结果图像中的部分显著区域进行了放大,如图 4所示.本文方法的融合结果图具有更理想的视觉效果.此外,直观效果在客观评价指标数据中也得到了验证,表 1表 2给出了5种融合算法的客观评价测度值.

图 4 五种融合方法的区域放大图 Fig. 4 Enlarged images based on five methods

表 1 第Ⅰ组灰度多聚焦图像融合效果性能比较 Tab. 1 Performance comparison of the grey multi-focus images in group Ⅰ

表 2 第Ⅱ组灰度多聚焦图像融合效果性能比较 Tab. 2 Performance comparison of the grey multi-focus images in group Ⅱ

如上所述,与现有代表性融合方法相比,本文提出的融合方法无论在直观视觉效果还是客观评价指标值方面均具有较明显的优势.

4 结论

本文针对多聚焦图像融合问题,提出了一种基于改进型CNN模型的图像融合方法.论述了多聚焦图像融合问题与CNN中分类问题的一致性关系,表明了本文方法的合理性和可行性;对经典的CNN模型加以改进,给出了权值设定方法,并对经典模型的池化层机制进行了改进;将待融合源图像输入改进型CNN模型进行了实验仿真.仿真结果验证了本文方法的有效性.

参考文献
[1]
DA C A, ZHOU J, DO M N. The nonsubsampled contourlet transform: theory, design, and applications[J]. IEEE transactions on image processing, 2006, 15(10): 3089-3101. DOI:10.1109/TIP.2006.877507 (0)
[2]
YANG Y, TONG S, HUANG S Y, et al. Multifocus image fusion based on NSCT and focused area detection[J]. IEEE sensors journal, 2015, 15(5): 2824-2838. (0)
[3]
LIM W Q. Nonseparable shearlet transform[J]. IEEE transactions on image processing, 2013, 22(5): 2056-2065. DOI:10.1109/TIP.2013.2244223 (0)
[4]
GAO G R, XU L P, FENG D Z. Multi-focus image fusion based on non-subsampled shearlet transform[J]. IET image processing, 2013, 7(6): 633-639. DOI:10.1049/iet-ipr.2012.0558 (0)
[5]
LUO X Q, ZHANG Z C, ZHANG B C, et al. Image fusion with contextual statistical similarity and nonsubsampled shearlet transform[J]. IEEE sensors journal, 2017, 17(6): 1760-1771. DOI:10.1109/JSEN.2016.2646741 (0)
[6]
赵杰, 王配配. 基于NSST变换域WNNM和KAD算法的SAR图像去噪[J]. 郑州大学学报(理学版), 2017, 49(2): 72-77. (0)
[7]
刘洪普, 杨乐, 侯向丹, 等. 一种改进的模糊C均值图像分割算法[J]. 郑州大学学报(理学版), 2017, 49(2): 66-71. (0)
[8]
YANG Y, TONG S, HUANG S Y, et al. A hybrid method for multi-focus image fusion based on fast discrete curvelet transform[J]. IEEE access, 2017, 5: 14898-14913. DOI:10.1109/ACCESS.2017.2698217 (0)
[9]
NEJATI M, SAMAVI S, SHIRANI S. Multi-focus image fusion using dictionary-based sparse representation[J]. Information fusion, 2015, 25(1): 72-84. (0)
[10]
YIN H P, LI Y X, CHAI Y, et al. A novel sparse-representation-based multi-focus image fusion[J]. Neurocomputing, 2016, 216(12): 216-229. (0)
[11]
靳霞, 雳志豪. 框架变换耦合SUSAN的多聚焦图像融合算法[J]. 计算机工程与设计, 2017, 38(9): 2425-2431. (0)
[12]
管飚. 基于小波变换的多聚焦图像融合方法[J]. 吉林大学学报(理学版), 2017, 55(4): 915-920. (0)
[13]
刘栓, 刘直良, 张权焯. 非下采样Contourlet变换耦合区域特性的多聚焦图像融合算法[J]. 河南理工大学学报(自然科学版), 2017, 36(4): 106-112. (0)
[14]
DU C B, GAO S S. Image segmentation-based multi-focus image fusion through multi-scale convolutional neural network[J]. IEEE access, 2017, 5: 15750-15761. DOI:10.1109/ACCESS.2017.2735019 (0)
[15]
LIU Z, BLASCH E, XUE Z Y, et al. Fusion algorithms for context enhancement in night vision: a comparative study[J]. IEEE transactions on pattern analysis and machine intelligence, 2012, 34(1): 94-109. DOI:10.1109/TPAMI.2011.109 (0)