当前位置: 首页 > 专利查询>中南大学专利>正文

基于卷积神经网络的图像融合方法技术

技术编号:28981953 阅读:3 留言:0更新日期:2021-06-23 09:29
本发明专利技术公开了一种基于卷积神经网络的图像融合方法,包括获取训练数据集;构建基于卷积神经网络的图像融合模型并训练得到图像融合模型;将待融合的两张图像输入到图像融合模型中完成图像的融合。本发明专利技术将图像配准和图像融合过程在一个网络中进行训练,并共享编码器,从而减少了网络参数,加速了网络的训练过程;同时设计了一种DenseBlock结构,使得特征提取过程更为精确和可靠;最后采用配准解码器网络和跳跃连接的方式,使得网络的最后输出的变形场能够捕获到浅层网络的特征信息,并且能够和深层网络特征进行融合;因此,本发明专利技术方法的可靠性高、实用性好且精确性较好。

【技术实现步骤摘要】
基于卷积神经网络的图像融合方法
本专利技术属于图像处理领域,具体涉及一种基于卷积神经网络的图像融合方法。
技术介绍
随着经济技术的发展,图像处理技术已经得到了广泛应用。而在进行图像处理时,经常涉及到图像的融合过程。图像融合后,融合后的图像既能够保持原图像中的重要信息,又能够反映出新增加的图像的信息的变换;因此,融合后的图像往往结合了来自多个图像的互补信息和冗余信息,能够提供更为丰富的数据信息。目前随着人工智能算法的普及,人工智能算法(比如卷积神经网络算法)也已经广泛应用于图像融合领域。但是,目前的基于人工智能算法的图像融合技术,一般是先训练图像配准网络并用于图像配准,然后再将配准后的图像进行图像融合网络训练,再进行图像融合。但是,明显的,现有的基于人工智能算法的图像融合技术,其训练过程较为繁琐,而且可靠性和精确性都不高。
技术实现思路
本专利技术的目的在于提供一种可靠性高、实用性好且精确性较好的基于卷积神经网络的图像融合方法。本专利技术提供的这种基于卷积神经网络的图像融合方法,包括如下步骤:S1.获取训练数据集;S2.构建基于卷积神经网络的图像融合模型;S3.采用步骤S1获取的训练数据集,对步骤S2构建的基于卷积神经网络的图像融合模型进行训练,从而得到图像融合模型;S4.将待融合的两张图像输入到步骤S3得到的图像融合模型中,完成图像的融合。步骤S1所述的训练数据集,具体包括浮动图像和参考图像。步骤S2所述的构建基于卷积神经网络的图像融合模型,具体为基于卷积神经网络的图像融合模型采用如下步骤进行图像融合:A.将浮动图像和参考图像输入到编码器网络中进行编码和特征提取;B.将步骤A提取到的特征进行串接,并将串接后的特征输入到配准解码器网络,从而得到配准参数;C.利用步骤B得到的配准参数,对浮动图像进行变换;D.将变换后的浮动图像输入到编码器网络进行编码;E.将步骤D得到的编码后的变换浮动图像与步骤A得到的编码后的参考图像输入到融合层进行融合;F.将步骤E得到的融合后的数据输入到重构解码器网络中,从而得到最终的融合后的图像。步骤A所述的编码器网络中,具体为采用DenseBlock结构作为编码器网络。所述的DenseBlock结构包括第一卷积核、第二卷积核和第三卷积核;第一卷积核、第二卷积核和第三卷积核依次串接;第一卷积核的大小为16*16*3*3*3;第二卷积核的大小为32*16*3*3*3;第三卷积核的大小为48*16*3*3*3;上述卷积核的参数定义为输入通道数*输出通道数*长*宽*高;第一卷积核的输入数据同时输入到第二卷积核的输入端、第三卷积核的输入端和第三卷积核的输出端;第一卷积核的输出数据同时输入到第三卷积核的输入端和第三卷积核的输出端;第二卷积核的输出数据输入到第三卷积核的输出端;第三卷积核的输出数据与第一卷积核的输入数据、第一卷积核的输出数据和第二卷积核的输出数据融合后,作为最终的DenseBlock结构的输出数据。步骤B所述的将步骤A提取到的特征进行串接,具体为将步骤A提取到的特征直接进行串接,从而实现特征的融合。步骤B所述的配准解码器网络,具体为采用类似U-net的结构作为配准解码器网络。所述的配准解码器网络,具体包括上采样模块和下采样模块;下采样模块包括第一下采样层、第二下采样层、第三下采样层和第四下采样层;上采样模块包括第一上采样层、第二上采样层、第三上采样层和第四上采样层;下采样模块的输入数据通过第一下采样层进行下采样后缩小1/2,第一下采样层的输出数据输入到第二下采样层,同时第一下采样层的输出数据也和第三上采样层的输出数据进行串接;第二下采样层将接收到的数据进行下采样后缩小1/2,第二下采样层的输出数据输入到第三下采样层,同时第二下采样层的输出数据也和第二上采样层的输出数据进行串接;第三下采样层将接收到的数据进行下采样后缩小1/2,第三下采样层的输出数据输入到第四下采样层,同时第三下采样层的输出数据也和第一上采样层的输出数据进行串接;第四下采样层将接收到的数据进行下采样后缩小1/2,并将数据输出到第一上采样层;第一上采样层将第四下采样层的输出数据进行上采样后放大两倍;第二上采样层将第一上采样层的输出数据和第三下采样的数据进行上采样后放大两倍;第二上采样层将第一上采样层的输出数据和第二下采样层得到的数据进行上采样后放大两倍;第三上采样层将第二上采样层的输出数据和第一下采样层得到的数据进行上采样后放大两倍;第四上采样层将第三上采样层的输出数据进行上采样后放大两倍;最后经过卷积层操作,从而得到最终的配准解码器网络的输出。所述的配准解码器网络,具体为下采样过程中,每个卷积核的大小为3*3*3,每次卷积操作后,均进行池化操作,具体为LeakyRelu;然后在上采样过程中,采用反卷积操作,并且每次反卷积之后,加上LeakyRelu的池化操作。步骤E所述的将步骤D得到的编码后的变换浮动图像与步骤A得到的编码后的参考图像输入到融合层进行融合,具体为采用如下算式进行融合:式中Ffused(x,y)为融合后的数据;Ef(x,y)为步骤A得到的编码后的参考图像;为步骤D得到的编码后的变换浮动图像。步骤F所述的重构解码器网络,具体包括第一卷积核、第二卷积核、第三卷积核和第四卷积核;第一卷积核、第二卷积核、第三卷积核和第四卷积核依次串联;第一卷积核的大小为64*64*3*3*3;第二卷积核的大小为64*32*3*3*3;第三卷积核的大小为32*16*3*3*3;第三卷积核的大小为16*1*3*3*3;参数定义为卷积核的输入通道数*输出通道数*长*宽*高。步骤S3所述的训练,具体为将图像配准和图像融合进行联合训练,同时在训练时,采用如下损失函数进行训练:式中Loss为损失函数;α、β和γ均为超参数;Lsim(.)为相似性度量函数,其中相似性度量函数采用NCC;Lsmo(.)为正则化项;F为输入的参考图像;为采用配准参数对浮动图像进行变换的结果;FD为重构编码器输出的图像;φ为配准参数,也即变形场。本专利技术提供的这种基于卷积神经网络的图像融合方法,将图像配准和图像融合过程在一个网络中进行训练,并共享编码器,从而减少了网络参数,加速了网络的训练过程;同时设计了一种DenseBlock结构,使得特征提取过程更为精确和可靠;最后采用配准解码器网络和跳跃连接的方式,使得网络的最后输出的变形场能够捕获到浅层网络的特征信息,并且能够和深层网络特征进行融合;因此,本专利技术方法的可靠性高、实用性好且精确性较好。附图说明图1为本专利技术方法的方法流程示意图。图2为本专利技术方法的配准及融合流程示意图。图3为本专利技术方法的Denseblock结构的结构示意图。图4为本专利技术方法的配准解码器网络的结构示意图。图5为本专利技术方法的重构解码器网络的结构示意图。具体实施方式如图1所示为本专利技术方法的方本文档来自技高网...

【技术保护点】
1.一种基于卷积神经网络的图像融合方法,包括如下步骤:/nS1.获取训练数据集;/nS2.构建基于卷积神经网络的图像融合模型;/nS3.采用步骤S1获取的训练数据集,对步骤S2构建的基于卷积神经网络的图像融合模型进行训练,从而得到图像融合模型;/nS4.将待融合的两张图像输入到步骤S3得到的图像融合模型中,完成图像的融合。/n

【技术特征摘要】
1.一种基于卷积神经网络的图像融合方法,包括如下步骤:
S1.获取训练数据集;
S2.构建基于卷积神经网络的图像融合模型;
S3.采用步骤S1获取的训练数据集,对步骤S2构建的基于卷积神经网络的图像融合模型进行训练,从而得到图像融合模型;
S4.将待融合的两张图像输入到步骤S3得到的图像融合模型中,完成图像的融合。


2.根据权利要求1所述的基于卷积神经网络的图像融合方法,其特征在于步骤S2所述的构建基于卷积神经网络的图像融合模型,具体为基于卷积神经网络的图像融合模型采用如下步骤进行图像融合:
A.将浮动图像和参考图像输入到编码器网络中进行编码和特征提取;
B.将步骤A提取到的特征进行融合,并将融合后的特征输入到配准解码器网络,从而得到配准参数;
C.利用步骤B得到的配准参数,对浮动图像进行变换;
D.将变换后的浮动图像输入到编码器网络进行编码;
E.将步骤D得到的编码后的变换浮动图像与步骤A得到的编码后的参考图像输入到融合层进行融合;
F.将步骤E得到的融合后的数据输入到重构解码器网络中,从而的得到最终的融合后的图像。


3.根据权利要求2所述的基于卷积神经网络的图像融合方法,其特征在于步骤A所述的编码器网络,具体为采用DenseBlock结构作为编码器网络;DenseBlock结构包括第一卷积核、第二卷积核和第三卷积核;第一卷积核、第二卷积核和第三卷积核依次串接;第一卷积核的大小为16*16*3*3*3;第二卷积核的大小为32*16*3*3*3;第三卷积核的大小为48*16*3*3*3;参数定义为输入通道数*输出通道数*长*宽*高;第一卷积核的输入数据同时输入到第二卷积核的输入端、第三卷积核的输入端和第三卷积核的输出端;第一卷积核的输出数据同时输入到第三卷积核的输入端和第三卷积核的输出端;第二卷积核的输出数据输入到第三卷积核的输出端;第三卷积核的输出数据与第一卷积核的输入数据、第一卷积核的输出数据和第二卷积核的输出数据融合后,作为最终的DenseBlock结构的输出数据。


4.根据权利要求3所述的基于卷积神经网络的图像融合方法,其特征在于步骤B所述的将步骤A提取到的特征进行融合,具体为将步骤A提取到的特征直接进行串接,从而实现特征的融合。


5.根据权利要求4所述的基于卷积神经网络的图像融合方法,其特征在于步骤B所述的配准解码器网络,具体为采用类似U-net结构作为配准解码器网络。


6.根据权利要求5所述的基于卷积神经网络的图像融合方法,其特征在于所述的配准解码器网络,具体包括上采样模块和下采样模块;下采样模块包括第一下采样层、第二下采样层、第三下采样层和第四下采样层;上采样模块包括第一上采样层、第二上采样层、第三上采样层和第四上采样层;下采样模块的输入数据通过第一下采样层进...

【专利技术属性】
技术研发人员:梁毅雄程海涛刘晴刘剑锋
申请(专利权)人:中南大学
类型:发明
国别省市:湖南;43

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1