光学仪器  2021, Vol. 43 Issue (5): 40-47   PDF    
基于多尺度特征融合网络的多聚焦图像融合技术
吕晶晶, 张荣福     
上海理工大学 光电信息与计算机工程学院,上海 200093
摘要: 多聚焦图像融合技术是为了突破传统相机景深的限制,将焦点不同的多幅图像合成一幅全聚焦图像,以获得更加全面的信息。以往基于空间域和基于变换域的方法,需要手动进行活动水平的测量和融合规则的设计,较为复杂。所提出的方法与传统的神经网络相比增加了提取浅层特征信息的部分,提高了分类准确率。将源图像输入训练好的多尺度特征网络中获得初始焦点图,然后对焦点图进行后处理,最后使用逐像素加权平均规则获得全聚焦融合图像。实验结果表明,本文方法融合而成的全聚焦图像清晰度高,保有细节丰富且失真度小,主、客观评价结果均优于其他方法。
关键词: 多聚焦图像融合    卷积神经网络    多尺度特征    
Multi-focus image fusion technology based on multi-scale feature fusion network
LÜ Jingjing, ZHANG Rongfu     
School of Optoelectronic Information and Computer Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China
Abstract: The multi-focus image fusion technology is to break through the limitation of the traditional camera's depth of field and combine multiple images with different focal points into a full-focus image to obtain more comprehensive information. In the past, methods based on the spatial domain and the transform domain required manual activity level measurement and fusion rule design, which was more complicated. Compared with the traditional neural network, the method proposed in this paper increases the part of extracting shallow feature information and improves the classification accuracy. The source image is input into the trained multi-scale feature network to obtain the initial focus map. Then, the focus map is post-processed . Finally , the pixel-by-pixel weighted average rule is used to obtain the all-focus fusion image. The experimental results show that the all-focus image fused by the method in this paper has high definition, rich details, and low distortion, the subjective and objective evaluation results are better than those of other methods for comparison.
Key words: multi-focus image fusion    convolutional neural network    multi-scale features    
引 言

图像融合技术是指将多张图像中的重要信息组合到一张图像中,比单一源图像具有更丰富的细节[1]。由于受到景深的限制,光学器件很难捕捉到画面中所有清晰的物体[2]。针对这一问题,本文提出了一种多聚焦图像融合算法,其目的是将焦点不同的图像组合成全聚焦图像。作为扩展光学透镜景深的一种有效且低成本的方法,多聚焦图像融合技术在生物学、医学、工业、农业等领域具有广阔的应用空间[3-4]

过去的二十年里,人们提出了很多种图像融合方法,这些方法大致可分为基于变换域和基于空间域的两大类方法。在这两类方法中,基于变换域的方法符合人的视觉特征,其中的多尺度变换域方法具有更好的特征表示能力,而变换域的选择和融合规则的设计是融合过程中最关键的两个因素。典型的变换域方法包括金字塔变换[5]、小波变换[6]、非下采样剪切变换[7]、稀疏表示[8]等。空间域融合可分为基于像素的融合方法[9]和基于块的融合方法[10]。基于像素的方法简单明了,能很好地保存图像细节,易于实现。但是,基于像素的融合算法只考虑局部邻域的焦点信息,可能会造成对比度失真或块状伪影。基于块的方法可以避免基于像素的融合算法的一些缺点。但是,块大小的选择是一个问题,因为当块选择较大时,不能保证聚焦区域的精度,而当块选择较小时,可能无法检测到重点区域。因此,块大小的选择可能会限制融合的性能。

由于具有优越的图像特征获取和信息分析能力,深度学习受到了广泛的关注,并在一些图像处理和计算机视觉任务中获得了巨大的突破[11]。本文使用多尺度特征融合网络代替传统的卷积神经网络,在增加网络宽度和深度的同时减少了参数量,提取到多层特征使系统对不同情况更具鲁棒性。图像融合的过程是将源图像输入本文训练的模型中,得到初始聚焦图。然后对得到的聚焦图进行初始分割、形态学算子和小区域滤波等后处理,生成最终的决策图,最后利用决策图和源图像获得融合图像。使用深度学习方法来解决多聚集图像融合任务具有很大的优势,不仅可以克服测量活动水平和人工设计融合规则的困难,更重要的是通过训练网络模型,联合生成活动水平测量和融合规则,大大地提高了融合图像的质量[3]

1 网络设计 1.1 卷积神经网络

卷积神经网络(CNN)是一种流行的深度学习网络模型[12],成功地应用于各种图像融合任务,并获得了不错的效果。在一般的CNN架构中都会有卷积层(Conv)、整流线性单元(ReLU)、池化层(子采样)和全连接层(FC)。在CNN中每个卷积层将一定数量的输入图像转换为一定数量的特征图,然后下一个卷积层通过使用一组过滤器的卷积运算将这一特征图量转换为另一特征图量。ReLU激活函数的作用也很重要,它能使网络稀疏,提高运算效率,还能引入非线性因素,使得神经网络能够更好地解决更加复杂的问题。ReLU表达式如下:

${y^i} = \max \left( {0,\sum \nolimits_i {K^{ij}} \otimes {x^i} + {b^j}} \right)$ (1)

式中: $ {K}^{ij} $ 为卷积核; $ {b}^{j} $ 为偏置; $ \otimes $ 为卷积运算。在卷积操作之后,为了防止过拟合还会使用池化层来降低特征维数。全连接层一般是CNN的最后一层,它是一个核大小为1×1的卷积层。其原理如图1所示。

图 1 CNN原理图 Figure 1 Schematic diagram of CNN
1.2 网络结构

本文将多聚焦图像融合的过程看作一个二分类问题[13],把图像融合中活动水平的测量看作特征提取,融合规则类似于一般分类任务中的分类器,所以采用CNN在理论上是可行的[3]。在本文提出的方法中,构建了一个多尺度融合的卷积神经网络结构,如图2所示。网络结构可以分为三个部分:多尺度特征提取、特征融合和分类。

图 2 多尺度融合的卷积神经网络 (MFCNN) 结构 Figure 2 Schematic diagram of MFCNN structure

在多尺度特征提取部分,网络有3个分支。中间的分支有3个卷积层、1个最大池化层和1个上采样。其中3个卷积层分别有64、128、256个滤波器,它们的卷积核大小都为1×1,步长为1×1,填充为1×1,此外每一个卷积层后面都连接着一个激活函数ReLU,增加神经网络模型的非线性。最大池化层的尺寸为2×2,步数为2×2,与平均池化相比最大池化引入了平移不变性,增加了网络的非线性。为了统一维度,采用双线性插值法进行上采样。卷积神经网络在浅层能够学习一些边缘和纹理特征,在深层学习一些语义和类别信息。为了充分利用卷积过程中产生的浅层和深层特征,本文把第一层卷积得到的浅层特征连接1×1卷积进行处理,这样可以做到在不丢失大量信息的情况下降低输入层的维数,从而避免过拟合。在第二个卷积层之后得到的特征也连接1×1卷积进行处理。在多尺度融合层,将前面提取到的多尺度特征进行concatenation操作及向量化拼接融合,得到大小为16×32×352的特征。在分类层中,全连接层会综合所提取的特征,生成向量直接送入Softmax层中进行分类。网络结构参数细节如表1所示。

表 1 网络结构参数细节 Table 1 Network structure details
2 数据集构建及训练

训练卷积神经网络需要大量的标记训练图像,这样才能学习到更多有用的特征。目前还没有公开的具有聚焦和离焦标记的图像数据库,因此本文考虑在开源数据集的基础上,创建一个包含正确标记的聚焦图像和离焦图像的训练数据集,以满足训练要求。本文利用Coco数据集中的2200多张高质量图像建立了训练数据集[3]。Coco数据集里的图像都是从自然中捕获的聚焦图像,一些随机选择的Coco数据集的图像样本如图3所示。

图 3 Coco数据集 Figure 3 Coco dataset

创建数据集时,首先将选取的图像转换为灰度图像,为了模拟出离焦情况的图片,每一个从Coco数据集中选择的图像都通过4个不同的高斯滤镜生成4个版本的模糊图像,标准差分别为9×9,11×11,13×13和15×15。接着把清晰图像和经过模糊处理的图像裁切成贴片上下相连组合到一起,操作过程如图4所示。贴片的尺寸选取也至关重要,如果贴片太小,很难学习可用的特性;如果贴片尺寸过大,一张训练贴片可能会同时包含聚焦和离焦部分,影响训练效果。

图 4 构建数据集的过程 Figure 4 The process of building a dataset

最终本文选择把贴片的大小定为16×16,然后分别在清晰和模糊图像中的对应位置取出16×16大小的贴片PA和PB垂直排列组合成16×32的贴片。与其他方法中单独将两个贴片分别投入网络中相比,本文将贴片对组合到一起投入网络中,可以提高分类的准确性。如果上面是清晰贴片下面是模糊贴片的,上焦点组合被标记为0,相反上面是模糊贴片下面是清晰贴片的,下焦点组合则被标记为1。这样便生成了本文训练所需要的数据集,包括100000对训练贴片和10000对验证贴片,示例贴片对如图5所示。

图 5 数据集示例 Figure 5 Example dataset

本方法中的网络使用随机梯度下降优化器(SGD)进行训练,这是深度学习网络中最常见的优化器。在本文的训练过程中,学习率指定为0.0001,批量大小为128个样本,动量为0.9,重量衰减为0.0005,更新权重规则如下:

$ {v}_{i+1}=0.9{v}_{i}-0.000\;5\mathrm{\alpha }{w}_{i}-\alpha \frac{{\mathrm{\delta }}{L}}{\delta {w}_{i}},\;{w}_{i+1}={w}_{i}+{v}_{i+1} $ (2)

式中: $ v $ 是动量变量; $ i $ 是迭代指标; $ \mathrm{\alpha } $ 是学习率;L是损失函数; $ \dfrac{\mathrm{\delta }{L}}{\mathrm{\delta }{{w}}_{{i}}} $ 是损失相对于权重 $ {w}_{i} $ 的导数。

3 设计融合方案

在进行图像融合处理之前,输入的图像已经通过图像预处理进行滤波降噪、校正配准,如果没有进行图像预处理,融合图像的质量将很难保证。如果需要融合两张以上的照片,可以采取串联的方式依次融合。输入的多聚焦图像是彩色图像,则需要将其转换为灰度图像,然后输入预训练好的网络中,网络会输出清晰度数值。融合步骤如图6所示。可以看出聚焦信息被准确地检测到,细节丰富的区域的清晰度值接近1(黑色)或0(白色),而平缓区域的值往往接近0.5(灰色)。由于网络中的跨步操作,网络输出的初始焦点图大小与原始图像不同。要生成与源图像大小相同的全聚焦图像,只需将每个值分配给其对应的色块,然后对重叠像素进行平均,即可获得。

图 6 图像融合的过程 Figure 6 The process of image fusion

在获得分数矩阵后,采用流行的选择最大值策略处理矩阵。本文将矩阵转换为二值化图,阈值定为0.5,其处理方法如下:

$ {D}\left({r},{c}\right)=\left\{\begin{array}{l}1,\;\;\;\;M\left(r,c\right) > 0.5\\ 0,\;\;\;\;{\rm{else}}\end{array}\right. $ (4)

式中: $ r,c $ 为图像矩阵中的坐标。二值化图中可能会包含一些错误分类的像素,为了解决这个问题,本文利用小区域过滤器清除孤立区域或填充小孔。另外本文使用形态过滤器中的腐蚀来处理过渡部分常常存在细线和毛刺的问题,从而使连接部分更加平滑,获得了比较满意的最终分割图 $ {D} $ 。最后按照逐像素加权平均规则,计算出融合图像 $ {F} $

$ {F}\left({r},{c}\right)={D}\left({r},{c}\right)\times {A}\left({r},{c}\right)+\left[1-{D}\left({r},{c}\right)\right]\times {B}\left({r},{c}\right) $ (5)

式中: $ {A}\left({r},{c}\right){\text{,}}{B}\left({r},{c}\right) $ 为输入的多聚焦图像。

4 实验与结果分析

为了验证所用方法的有效性,本文将提出的方法与影响力较高的一些方法进行了主、客观两方面的比较,参与对比的方法包括基于块(sssdi)的方法,基于稀疏表示(sr)的方法,基于多尺度分解(gff)的方法,基于像素(misf)的方法,基于卷积神经网络(cnn)的方法。

4.1 客观评价

客观评价在图像融合效果的评价中起着重要作用,融合效果主要由多个指标的定量得分来综合评估。Liu等[3]将常用的指标分为4类:基于信息论,基于图像特征,基于图像结构相似性和基于人类感知。在这项研究中,本文从每个类别中选择一个指标进行评估。选择的4个度量是:归一化互信息 $ {Q}_{{\rm{MI}}} $ [14]、基于梯度的指标 $ {Q}_{{\rm{G}}} $ [15]、基于结构相似性的度量 $ {Q}_{{\rm{Y}}} $ [16]、基于人类感知的度量 $ {Q}_{{\rm{CB}}} $ [17]。其中指标 $ {Q}_{{\rm{MI}}} $ 表示融合图像和源图像之间的互信息量, $ {Q}_{{\rm{G}}} $ 表示从源图像注入融合图像的空间细节的程度, $ {Q}_{{\rm{Y}}} $ 表示源图像和融合图像之间的感知差异, $ {Q}_{{\rm{CB}}} $ 表示在融合图像中的源图像的对比度信息。

表2表3分别描述的是20对彩色图像和10对灰色图像,采用6种融合方法质量评估指标的平均值,数值较大表示有较好的性能。虽然本文所提出的方法无论是彩色图像融合还是灰度图像融合实验,指标 $ {Q}_{{\rm{MI}}} $ 都略微落后,但是从4个客观评价指标综合来看,本文方法在客观指标评价方面优于其他方法。

表 2 彩色图像评价指标对比 Table 2 Comparison of color image evaluation indicators

表 3 灰度图像评价指标对比 Table 3 Comparison of gray image evaluation indicators
4.2 主观评价

主观评价方法是评价者通过人眼视觉感知来对融合图像进行评价,主要的评价标准有:1) 最终融合图像的清晰度和对比度是否和源图像中的清晰区域保持一致,是否存在图像质量的严重降低;2) 融合图像中离焦和聚焦结合区域是否存在伪影、晕影之类的不清晰情况。

图7展示了lytro数据集中的“model girl”图像对的6种方法融合结果。从图中可以看出,由于图像中靠近肩部区域的背景区域中存在不希望出现的噪声,因此gff方法的融合结果非常差。sssdi、misf和sr方法的融合结果比gff方法更好,但是仍存在一些多余的模糊像素。基于cnn的方法可产生良好的融合图像,然而边界周围包含一些模糊伪影。本文所提出的方法产生的融合结果非常清晰,轮廓鲜明,并且几乎不包含边界附近的模糊像素。因此,在视觉主观感知方面,本文提出的方法保留了边缘信息并生成了比其他图像融合方法更好的融合图像。

图 7 6种彩色图像融合方法的对比 Figure 7 Comparison of the results of the six color image fusion methods

图8是来自“clock”图像对的6种方法融合结果。其中sr方法生成的结果图整体质量较差,画质模糊。sssdi、cnn比sr方法生成的结果图的质量稍好,不过在这两种方法的结果图中,小时钟与大时钟交叉部分轮廓都有部分模糊的情况,有分割不准确的问题。gff方法生成的图像较为完整,美中不足在于大时钟边框有些许伪影。misf方法和本文提出的方法生成的图像清晰度高,精准提取聚焦区域,细节保留较完整。

图 8 6种灰度图像融合方法的对比 Figure 8 Comparison of the results of the six gray image fusion methods
5 结论与展望

多聚焦图像融合的过程可以概括为从源图像中提取聚焦区域的信息,然后将这些信息组合形成全聚焦图像。本文使用深度学习的方法来解决多聚焦图像融合任务,采用多级特征融合的手段,将浅层提取的信息和深层提取的信息相结合,充分利用提取到的特性,从而减少了边界上的模糊伪影,提高了融合图像的清晰度。实验结果表明,与现有技术相比,该方法在主观评价和客观评价方面均有很大的提高。后续工作可以将此技术应用到具体的场景中,比如显微镜操作平台,可以达到扩展显微成像设备景深的目的。

参考文献
[1] 杨艳春, 李娇, 党建武, 等. 基于引导滤波与改进PCNN的多聚焦图像融合算法[J]. 光学学报, 2018, 38(5): 0510001.
[2] 邹爽, 许忠保, 吕清花. 大景深显微成像方法研究[J]. 光电工程, 2013, 40(5): 120–126.
[3] LIU Y, CHEN X, PENG H, et al. Multi-focus image fusion with a deep convolutional neural network[J]. Information Fusion, 2017, 36: 191–207. DOI:10.1016/j.inffus.2016.12.001
[4] 华东师范大学, 上海澜澈生物科技有限公司. 一种基于小波变换的多聚焦显微图像融合方法: CN 201810938548.9[P]. 2019-02-01.
[5] WANG W, CHANG F. A multi-focus image fusion method based on Laplacian pyramid[J]. Journal of Computers, 2011, 6(12): 2559–2566.
[6] 朱世松, 瞿佩云. 基于小波变换和引导滤波的多聚焦图像融合[J]. 测控技术, 2020, 39(9): 103–107.
[7] RAMLAL S D, SACHDEVA J, AHUJA C K, et al. Multimodal medical image fusion using non-subsampled shearlet transform and pulse coupled neural network incorporated with morphological gradient[J]. Signal, Image and Video Processing, 2018, 12(8): 1479–1487. DOI:10.1007/s11760-018-1303-z
[8] LI X H, SHEN H F, ZHANG L P, et al. Recovering quantitative remote sensing products contaminated by thick clouds and shadows using multitemporal dictionary learning[J]. IEEE Transactions on Geoscience and Remote Sensing, 2014, 52(11): 7086–7098. DOI:10.1109/TGRS.2014.2307354
[9] ASLANTAS V, TOPRAK A N. A pixel based multi-focus image fusion method[J]. Optics Communications, 2014, 332: 350–358. DOI:10.1016/j.optcom.2014.07.044
[10] QIU X H, LI M, ZHANG L Q, et al. Guided filter-based multi-focus image fusion through focus region detection[J]. Signal Processing: Image Communication, 2018, 72: 35–46.
[11] 陈先昌. 基于卷积神经网络的深度学习算法与应用研究[D]. 杭州: 浙江工商大学, 2014.
[12] LIU Y, WANG L, CHENG J, et al. Multi-focus image fusion: a survey of the state of the art[J]. Information Fusion, 2020, 64: 71–91. DOI:10.1016/j.inffus.2020.06.013
[13] 翟浩, 庄毅. 利用拉普拉斯能量和CNN的多聚焦图像融合方法[J]. 哈尔滨工业大学学报, 2020, 52(5): 137–147. DOI:10.11918/201909064
[14] HOSSNY M, NAHAVANDI S, CREIGHTON D. Comments on 'information measure for performance of image fusion'[J]. Electronics Letters, 2008, 44(18): 1066–1067. DOI:10.1049/el:20081754
[15] XYDEAS C S, PETROVIĆ V. Objective image fusion performance measure[J]. Electronics Letters, 2000, 36(4): 308–309. DOI:10.1049/el:20000267
[16] YANG C, ZHANG J Q, WANG X R, et al. A novel similarity based quality metric for image fusion[J]. Information Fusion, 2008, 9(2): 156–160. DOI:10.1016/j.inffus.2006.09.001
[17] CHEN Y, BLUM R S. A new automated quality assessment algorithm for image fusion[J]. Image and Vision Computing, 2009, 27(10): 1421–1432. DOI:10.1016/j.imavis.2007.12.002