CN111161201A - 基于细节增强通道注意力的红外与可见光图像融合方法 - Google Patents
基于细节增强通道注意力的红外与可见光图像融合方法 Download PDFInfo
- Publication number
- CN111161201A CN111161201A CN201911338674.1A CN201911338674A CN111161201A CN 111161201 A CN111161201 A CN 111161201A CN 201911338674 A CN201911338674 A CN 201911338674A CN 111161201 A CN111161201 A CN 111161201A
- Authority
- CN
- China
- Prior art keywords
- image
- infrared
- visible light
- information
- feature map
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Granted
Links
- 238000007500 overflow downdraw method Methods 0.000 title claims abstract description 35
- 230000004913 activation Effects 0.000 claims abstract description 39
- 238000010586 diagram Methods 0.000 claims abstract description 39
- 238000000034 method Methods 0.000 claims abstract description 38
- 230000004927 fusion Effects 0.000 claims abstract description 37
- WHHGLZMJPXIBIX-UHFFFAOYSA-N decabromodiphenyl ether Chemical compound BrC1=C(Br)C(Br)=C(Br)C(Br)=C1OC1=C(Br)C(Br)=C(Br)C(Br)=C1Br WHHGLZMJPXIBIX-UHFFFAOYSA-N 0.000 claims abstract description 11
- 238000011176 pooling Methods 0.000 claims description 30
- 238000012549 training Methods 0.000 claims description 11
- 230000002708 enhancing effect Effects 0.000 claims description 9
- 238000013527 convolutional neural network Methods 0.000 claims description 8
- 230000003213 activating effect Effects 0.000 claims description 7
- 238000012935 Averaging Methods 0.000 claims description 6
- 238000013528 artificial neural network Methods 0.000 claims description 6
- 230000009191 jumping Effects 0.000 claims description 5
- 238000012360 testing method Methods 0.000 claims description 4
- 238000011478 gradient descent method Methods 0.000 claims description 3
- 238000013507 mapping Methods 0.000 claims description 3
- 238000007781 pre-processing Methods 0.000 claims description 3
- 230000009467 reduction Effects 0.000 claims description 3
- 238000000605 extraction Methods 0.000 abstract description 2
- 230000006870 function Effects 0.000 description 47
- 230000000694 effects Effects 0.000 description 10
- 238000013135 deep learning Methods 0.000 description 4
- 238000001514 detection method Methods 0.000 description 4
- 238000013441 quality evaluation Methods 0.000 description 4
- 230000009286 beneficial effect Effects 0.000 description 2
- 238000007796 conventional method Methods 0.000 description 2
- 230000008569 process Effects 0.000 description 2
- 241001465754 Metazoa Species 0.000 description 1
- 238000000354 decomposition reaction Methods 0.000 description 1
- 230000007547 defect Effects 0.000 description 1
- 238000002059 diagnostic imaging Methods 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 230000006872 improvement Effects 0.000 description 1
- 230000014759 maintenance of location Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000003058 natural language processing Methods 0.000 description 1
- 238000011002 quantification Methods 0.000 description 1
- 238000011158 quantitative evaluation Methods 0.000 description 1
- 238000013139 quantization Methods 0.000 description 1
- 238000011160 research Methods 0.000 description 1
- 238000004088 simulation Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T5/00—Image enhancement or restoration
- G06T5/50—Image enhancement or restoration using two or more images, e.g. averaging or subtraction
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/10—Image acquisition modality
- G06T2207/10048—Infrared image
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20084—Artificial neural networks [ANN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06T—IMAGE DATA PROCESSING OR GENERATION, IN GENERAL
- G06T2207/00—Indexing scheme for image analysis or image enhancement
- G06T2207/20—Special algorithmic details
- G06T2207/20212—Image combination
- G06T2207/20221—Image fusion; Image merging
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Life Sciences & Earth Sciences (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Health & Medical Sciences (AREA)
- Image Processing (AREA)
- Image Analysis (AREA)
Abstract
本发明涉及基于细节增强通道注意力的红外与可见光图像融合方法,属于特征提取与图像融合技术领域。包括:1将红外图像和可见光图像剪裁成大小相等的图像块并级联;2将级联图像依次通过一个卷积层,一个批正则化层,一个激活函数,得到高维特征图;3将高维特征图通过DECA模块,得到信息增强的高维特征图;4重复Q次步骤2和3,进一步对高维特征图进行信息增强;5将信息进一步增强的高维特征图通过一个卷积层,一个批正则化层,一个激活函数运算,输出融合特征图;6计算损失函数,更新卷积层和全连接层参数;7固定卷积层和全连接层的参数,融合图像。所述方法能在融合图像中提取亮度信息,突出图像中的高亮目标,同时保留大量细节信息。
Description
技术领域
本发明涉及基于细节增强通道注意力的红外与可见光图像融合方法,属于特征提取与图像融合技术领域。
背景技术
图像融合技术是将来自不同传感器/模态的同一场景的数据组合到单个图像的过程。融合后的图像可以比源图像携带更全面的信息。针对不同的应用,例如计算机视觉,摄影和医学成像,已经提出了大量的图像融合方法。基于红外和可见光图像的融合图像可以被用于目标识别或检测,如何使融合图像既包含可见光图像中的细节和纹理信息,又包含红外图像中的高亮目标,从而尽可能多地将多模态信息保留在一幅图像中是目前研究的主要问题。
现有传统方法首先分解源图像,然后根据融合规则融合相应的系数,最后将融合的系数映射回图像域以获得融合的图像。由于针对不同图像使用的融合方法固定,分解工具相同,因此传统方法可能无法获得最佳融合效果。此外,手动设计的融合规则是复杂且繁琐的。
近年来,由于深度学习在许多领域(包括计算机视觉,语音识别,自然语言处理等)中的出色表现,这一方法也被应用到图像融合领域。通过使用深度学习,我们可以在大型训练数据集上训练深且复杂的网络,以提取各种特征并自动获取更通用的融合规则。训练完成后,可以快速地融合图像。现有的基于深度学习的图像融合方法主要在损失函数和网络结构两方面对模型进行改进,以达到更理想的融合效果,即在一幅图上保留更丰富的信息并突出红外目标。
发明内容
本发明的目的在于针对现有红外和可见光图像融合方法存在的细节纹理信息丢失、红外探测目标不明显等技术缺陷,提出了基于细节增强通道注意力的红外与可见光图像融合方法。
所述红外和可见光图像融合方法,包括如下步骤:
步骤一:对训练集中的红外图像和可见光图像进行预处理,生成数量及大小相等的红外图像块与可见光图像块;
步骤一具体为:从公开网站中下载数据集,数据集中包括m+d对红外图像和可见光图像;从数据集中随机选取m对红外图像和可见光图像作为训练集,并对红外图像编号R1到Rm,对可见光图像编号V1到Vm;剩下的d对红外图像和可见光图像作为测试集,对红外图像编号RTq,对可见光图像编号VTq;q的取值范围为1到d;将训练集中的所有红外图像和可见光图像进行剪裁,生成数量及大小相等的红外图像块和可见光图像块并编号;
其中,红外图像块按照从R1到Rm且每幅图像中从上到下及从左到右的顺序编号为r1到rn;将可见光图像块按照从V1到Vm且每幅图像中从上到下及从左到右的顺序编号为v1到vn;红外图像块总数以及可见光图像块总数均为n;
步骤二:将步骤一生成的数量及大小相等的红外图像块与可见光图像块按顺序分别级联,得到级联图像块;
其中,按顺序分别级联是指将红外图像块rp与可见光图像块vp级联并将级联后的图像块记为fp;p的取值范围为1到n,初始化p=1;
步骤三:将步骤二得到的级联图像块fp输入一个卷积层,一个批正则化层,一个激活函数映射到高维空间,提取特征,得到高维特征图;
其中,卷积层包括卷积核,卷积核的大小为3×3到5×5之间,激活函数使用LeakyRELU函数;高维特征图的维度为C;
步骤四:将步骤三得到的高维特征图通过一个通道注意力模块,即DECA模块,得到信息增强的高维特征图;
其中,DECA模块包含分支A和分支B;分支A用于增强高维特征图的亮度信息,分支B用于增强高维特征图的细节信息,分支A和分支B是并行结构;
分支A中包括全局平均池化单元A1、全连接单元A1、激活单元A1、全连接单元A2和激活单元A2;
分支B包括梯度运算单元、全局平均池化单元B1、全连接单元B1、激活单元B1、全连接单元B2和激活单元B2;
步骤四包含如下子步骤:
步骤4.1A:将步骤三得到的高维特征图输入分支A,通过全局平均池化a1得到含有亮度信息的高维特征图,再将含有亮度信息的高维特征图通过全连接a1和激活函数a1运算,得到突出亮度信息的高维特征图,再将突出亮度信息的高维特征图通过全连接a2和激活函数a2运算,得到尺度向量sb,跳至步骤4.2;
其中,全局平均池化a1在全局平均池化单元A1中实现,全连接a1在全连接单元A1中实现,激活函数a1使用RELU函数在激活单元A1中实现,全连接a2在全连接单元A2中实现,激活函数a2使用sigmoid函数在激活单元A2中实现;sb的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的亮度信息;
其中,全局平均池化a1的操作如式(1):
Xc表示步骤三得到的高维特征图的第c个通道,c的取值范围为1到C,Fsq(Xc)表示含有亮度信息的高维特征图的第c个通道,H、W分别为输入的步骤三得到的高维特征图的高和宽,Xc(i,j)表示Xc第i行和第j列处的值;
步骤4.1B:将步骤三得到的高维特征图输入分支B,通过梯度运算得到梯度高维特征图,再将梯度高维特征图通过全局平均池化b1、全连接b1和激活函数b1运算,得到突出梯度信息的高维特征图,再将突出梯度信息的高维特征图通过全连接b2和激活函数b2运算,得到尺度向量sg;
其中,梯度运算在梯度运算单元中实现,全局平均池化b1在全局平均池化单元B1中实现,全连接b1使用RELU函数在全连接单元B1中实现,激活函数b1在激活单元B1中实现,全连接b2在全连接单元B2中实现,激活函数b2使用sigmoid函数在激活单元B2中实现;sg的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的细节信息;
其中,全局平均池化b1的操作如式(1),梯度运算如式(2)所示:
其中,Fgrad(Xc)表示梯度高维特征图的第c个通道,c的取值范围为1到C;Xc(i,j-1)表示Xc第i行和第j-1列处的值;Xc(i-1,j)表示Xc第i-1行和第j列处的值;
步骤4.2:将sg和sb相加,得到尺度向量s,即(3):
s=ksb+sg (3)
其中,参数k用来平衡sg和sb的权重,k的取值范围在0.01到0.5之间;
步骤4.3:将步骤4.2中的尺度向量s与步骤三得到的高维特征图对应通道相乘,生成信息增强的高维特征图;
步骤五:将步骤四中得到的信息增强的高维特征图作为输入,重复Q次步骤三和步骤四,对信息增强的高维特征图进行进一步的信息提取和增强,得到信息进一步增强的高维特征图;
其中,Q的取值范围为1到5;
步骤六:将步骤五输出的信息进一步增强的高维特征图通过一个卷积层,一个批正则化层,一个激活函数运算,进行特征融合,输出融合特征图;
其中,卷积核的大小为3×3到5×5之间,激活函数使用Leaky RELU函数;
步骤七:将步骤六得到的融合特征图通过一个卷积核大小为1×1的卷积层,一个批正则化层,一个Tanh激活函数运算,进行降维映射,输出融合图像块;
至此,步骤三到步骤七构成了卷积神经网络F的操作;
步骤八:计算步骤七得到的融合图像块与步骤三级联图像块fp对应的红外图像块rp和可见光图像块vp之间的损失,并利用梯度下降方法回传损失,使损失函数的输出值最小化,更新卷积神经网络F中所有卷积层和全连接层的参数;
其中,损失函数如式(4)所示:
其中,L表示损失函数的输出值,If,rp,vp分别表示步骤七输出的融合图像块、组成步骤三中级联图像块的红外图像块和可见光图像块;||·||F表示Frobenius范数,超参数λ是调节权重的参数,取值范围在3到10之间;超参数γ是调节权重的参数,其取值范围在0.5到1.2之间;为融合图像块与红外图像块之差的Frobenius范数的平方,为融合图像块与可见光图像块之差的Frobenius范数的平方;为梯度约束条件,用于保留融合图像块的细节信息,表示梯度算子;
步骤九:判断p是否等于n,若是,则表明已得到新的卷积神经网络F中的所有卷积层和全连接层的参数,跳至步骤十;若否取p=p+1,跳至步骤三;
步骤十:固定步骤九中新的卷积神经网络F中的所有卷积层和全连接层的参数,依次将步骤一中的RTq和VTq级联得到级联图像Fq;
其中,q的取值范围为1到d;
步骤十一:依次输入级联图像F1到Fd,重复步骤三到七,输出融合图像P1到Pd;
至此,经过步骤一到步骤十一,完成了基于细节增强通道注意力的红外与可见光图像融合方法。
有益效果
本发明所述的基于细节增强通道注意力的红外与可见光图像融合方法,与现有融合红外和可见光图像的方法相比,具有如下有益效果:
1、所述红外与可见光图像融合方法与其他基于深度学习一类的红外与可见光图像融合方法相比,能够有效提取可见光图像和红外图像的亮度信息,保留图像中的高亮目标,为图像识别与检测等应用提供帮助;
2、所述方法通过引入DECA模块使网络具有通道选择功能并引入梯度信息,强调有价值的信息,实现了对源图像中细节信息的保留。
附图说明
图1为本发明基于细节增强通道注意力的红外与可见光图像融合方法及实施例中的流程示意图;
图2为本发明基于细节增强通道注意力的红外与可见光图像融合方法步骤四中DECA模块的结构示意图;
图3为本发明基于细节增强通道注意力的红外与可见光图像融合方法步骤三到步骤七中构成的卷积神经网络F的结构示意图;
图4为本发明所述方法的融合效果图;
图5为本发明所述方法的融合效果图的局部放大效果。
具体实施方式
下面结合附图并举实施例,对本发明基于细节增强通道注意力的红外与可见光图像融合方法进行详细描述。
实施例1
本实施例叙述了使用本发明所述的一种基于细节增强通道注意力的红外与可见光图像融合方法的具体实施。
在红外与可见光图像融合问题中,需要将两种源图像中包含的信息合并到一幅图像中。其中,红外图像通过亮度突出目标,如飞机、行人、动物等;可见光图像则包含更多的细节信息,如树木的枝叶、地砖的纹理等。图像融合方法将红外图像的高亮区域与可见光图像的细节信息同时保留到一幅图中,以便更好地进行观察。
本实例使用NVIDIAGeForceGTX1080TiGPU,python2.7编程环境,tensorflow框架。使用信息熵(IE),结构相似度(SSIM),边缘融合质量指标(QE),加权融合质量评价指标(QAB/F),差异相关度(SCD),特征互信息(FMIω,FMIdct)和多尺度结构相似度(MS_SSIM)作为融合质量评价指标。
图1是本发明具体实施方式中红外与可见光图像融合方法的流程图。
图2为本发明具体实施方式中基于细节增强通道注意力的红外与可见光图像融合方法步骤四中DECA模块的结构示意图,其中平衡参数k的值为0.1。
图3为本发明具体实施方式中基于细节增强通道注意力的红外与可见光图像融合方法步骤三到步骤七中构成的卷积神经网络F的结构示意图,共包含5个卷积层,前2层的卷积核大小为5×5,3到4层的卷积核大小为3×3,最后一层的卷积核大小为1×1,卷积步长均为1;每一层都进行批正则化;前4层选择Leaky Relu作为激活函数,最后一层使用Tanh作为激活函数;前3层末端各包含一个DECA模块。
图4是实验效果图,将本发明所述的融合方法与LP、CVT、DTCWT、SR、fusionGAN和RCGAN方法融合效果进行对比,图中第一行到最后一行依次表示红外图像、可见光图像、通过LP、CVT、DTCWT、SR、fusionGAN、RCGAN和本发明中所述方法融合得到的7组图像。
图5为图4实验效果图的局部放大图,具体来说,图5中从上到下及从左到右依次为图4中第3列第4行到第9行图片,分别为使用CVT、DTCWT、SR、fusionGAN、RCGAN和本发明所述方法得到的融合图像,由白色方框内的区域可以看出,与CVT、DTCWT、SR、fusionGAN和RCGAN方法相比,本方法在保留方框上部目标人物的同时更多地保留了树枝地细节,使图像更为清晰。
图5仅为图4实验效果的一个放大示例,图4中其他融合图像均满足类似结论,即高亮区域(如人、轮胎、飞机)均被保留,同时细节信息(如树枝、地面砖块边缘、房屋轮廓和窗户)更加清晰;表明了本发明所述方法在突出目标和保留细节方面有更好的效果。
采用本发明所述的基于细节增强通道注意力的红外与可见光图像融合方法对一组可见光与红外图像进行融合仿真,输出的融合结果具有如下量化指标:
表1基于细节增强通道注意力的红外与可见光图像融合方法的融合结果的量化指标
其中,量化指标选取信息熵(IE),结构相似度(SSIM),边缘融合质量指标(QE),加权融合质量评价指标(QAB/F),差异相关度(SCD),特征互信息(FMIω,FMIdct)和多尺度结构相似度(MS_SSIM)这8种融合质量评价指标。
表1中从上到下依次为通过LP、CVT、DTCWT、SR、fusionGAN、RCGAN和本发明所述方法融合得到融合图像的量化评价指标。由表1可以看出,相比于LP、CVT、DTCWT和fusionGAN方法,本发明所述方法在各项指标上均表现最佳,虽然SR和RCGAN分别在IE和QE上略高于本发明所述方法,但这两种方法仅在单一指标上表现出优势,其余指标均低于本发明所述方法;综上所述,本发明所述方法的性能指标为最佳。
本实例采用本发明提出的基于细节增强通道注意力的红外与可见光图像融合方法对TNO数据集中的图像进行融合,并与LP、CVT、DTCWT、SR、fusionGAN和RCGAN方法进行对比,体现了本发明所述方法的有益效果。
具体包括如下步骤:
步骤一:对红外图像和可见光图像进行预处理:从公开网站中下载TNO数据集,从数据集中随机选取40对红外和可见光图像作为训练集,并对红外图像编号R1到R40,对可见光图像编号V1到V40;将这40对红外和可见光图像从TNO中去除,再从TNO中剩下的图片中随机选取7对红外和可见光图像对作为测试集,并对红外图像编号RT1到RT7,对可见光图像编号VT1到VT7;将训练集中的所有红外图像和可见光图像进行剪裁,生成120×120大小的57227对红外图像块和可见光图像块并编号;
其中,红外图像块按照从R1到R40且每幅图像中从上到下及从左到右的顺序编号为r1到r57227;同理,可见光图像块按照从V1到V40且每幅图像中从上到下及从左到右的顺序编号为v1到v57227;
步骤二:将步骤一生成的数量及大小相等的红外图像块与可见光图像块按顺序分别级联,得到级联图像块;
其中,按顺序分别级联是指将红外图像块rp与可见光图像块vp级联并将级联后的图像块记为fp;p的取值范围为1到57227,初始化p=1;
步骤三:将步骤二得到的级联后的图像块fp输入一个卷积层,一个批正则化层,一个激活函数映射到高维空间,提取特征,得到高维特征图;
其中,卷积层包括卷积核,卷积核的大小为5×5,激活函数使用LeakyRELU函数;
步骤四:将步骤三得到的高维特征图通过一个通道注意力模块,即DECA模块,得到信息增强的高维特征图;
其中,DECA模块包含分支A和分支B;分支A用于增强高维特征图的亮度信息,分支B用于增强高维特征图的细节信息,A和B是并行结构;
分支A中包括全局平均池化单元A1、全连接单元A1、激活单元A1、全连接单元A2和激活单元A2;
分支B包括梯度运算单元、全局平均池化单元B1、全连接单元B1、激活单元B1、全连接单元B2和激活单元B2;
步骤四具体包含如下子步骤:
步骤4.1A:将步骤三得到的高维特征图输入分支A,通过全局平均池化a1得到含有亮度信息的高维特征图,再将含有亮度信息的高维特征图通过全连接a1和激活函数a1运算,得到突出亮度信息的高维特征图,再将该高维特征图通过全连接a2和激活函数a2运算,得到尺度向量sb;跳至步骤4.2;
其中,全局平均池化a1在全局平均池化单元A1中实现,全连接a1在全连接单元A1中实现,激活函数a1使用RELU函数在激活单元A1中实现,全连接a2在全连接单元A2中实现,激活函数a2使用sigmoid函数在激活单元A2中实现;sb的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的亮度信息;
其中,全局平均池化a1的操作公式如式(1);
步骤4.1B:将步骤三得到的高维特征图输入分支B,通过梯度运算得到梯度高维特征图,再将梯度高维特征图通过全局平均池化b1、全连接b1和激活函数b1运算,得到突出梯度信息的高维特征图,再将该高维特征图通过全连接b2和激活函数b2运算,得到尺度向量sg;
其中,梯度运算在梯度运算单元中实现,全局平均池化b1在全局平均池化单元B1中实现,全连接b1使用RELU函数在全连接单元B1中实现,激活函数b1在激活单元B1中实现,全连接b2在全连接单元B2中实现,激活函数b2使用sigmoid函数在激活单元B2中实现;sg的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的细节信息;
其中,全局平均池化b1的操作如式(1),梯度运算如式(2);
步骤4.2:将sg和sb相加,得到尺度向量s,即式(3);
其中,参数k的值为0.1;
步骤4.3:将步骤4.2中的尺度向量s与步骤三得到的高维特征图对应通道相乘,生成信息增强的高维特征图;
步骤五:将步骤四中得到的信息增强的高维特征图作为输入,重复2次步骤三和步骤四,对信息增强的高维特征图进行进一步的信息提取和增强,得到信息进一步增强的高维特征图;
其中,步骤三中卷积核的大小在第一次重复时为5×5,在第二次重复时为3×3;
步骤六:将步骤五输出的信息进一步增强的高维特征图通过一个卷积层,一个批正则化层,一个激活函数运算,进行特征融合,输出融合特征图;
其中,卷积核的大小为3×3,激活函数使用LeakyRELU函数;
步骤七:将步骤六得到的融合特征图通过一个卷积核大小为1×1的卷积层,一个批正则化层,一个Tanh激活函数运算,进行降维映射,输出融合图像块;
至此,步骤三到步骤七构成了卷积神经网络F的操作;
步骤八:计算步骤七得到的融合图像块与步骤三级联图像块fp对应的红外图像块rp和可见光图像块vp之间的损失,并利用梯度下降方法回传损失,使损失函数的输出值最小化,更新卷积神经网络F中所有卷积层和全连接层的参数;
其中,损失函数如式(4),超参数λ取值为5;超参数γ取值为0.8;;
步骤九:依次取p=2到p=57227,重复步骤三到八,得到新的卷积神经网络F中的所有卷积层和全连接层的参数;
步骤十:固定步骤九中新的积神经网络F中的所有卷积层和全连接层的参数,依次将步骤一中的RTq和VTq级联得到级联图像Fq;q取1到7;
步骤十一:依次输入级联图像F1到F7,重复步骤三到七,输出融合图像P1到P7;
至此,经过步骤一到步骤十一,就完成了融合可见光和红外图像的全部过程;经实验验证,该方法能有效融合可见光图像和红外图像,并且能够在突出目标的同时保留大量细节和纹理信息,从而更好地辅助图像识别和目标检测等任务。
综上所述,以上仅为本发明的较佳实施例而已,并非用于限定本发明的保护范围。凡在本发明的精神和原则之内,所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (10)
1.基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:包括如下步骤:
步骤一:对训练集中的红外图像和可见光图像进行预处理,生成数量及大小相等的红外图像块与可见光图像块;
其中,训练集中包含m对红外图像和可见光图像;红外图像块按照从R1到Rm且每幅图像中从上到下及从左到右的顺序编号为r1到rn;将可见光图像块按照从V1到Vm且每幅图像中从上到下及从左到右的顺序编号为v1到vn;红外图像块总数以及可见光图像块总数均为n;对红外图像编号RTq,对可见光图像编号VTq;q的取值范围为1到d;d为测试集中包含红外图像和可见光图像对的数量;
步骤二:将步骤一生成的数量及大小相等的红外图像块与可见光图像块按顺序分别级联,得到级联图像块;
其中,按顺序分别级联是指将红外图像块rp与可见光图像块vp级联并将级联后的图像块记为fp;初始化p=1;
步骤三:将步骤二得到的级联图像块fp输入一个卷积层,一个批正则化层,一个激活函数映射到高维空间,提取特征,得到高维特征图;
其中,卷积层包括卷积核;
步骤四:将步骤三得到的高维特征图通过一个通道注意力模块,即DECA模块,得到信息增强的高维特征图;
其中,DECA模块包含分支A和分支B;分支A用于增强高维特征图的亮度信息,分支B用于增强高维特征图的细节信息,分支A和分支B是并行结构;
分支A中包括全局平均池化单元A1、全连接单元A1、激活单元A1、全连接单元A2和激活单元A2;
分支B包括梯度运算单元、全局平均池化单元B1、全连接单元B1、激活单元B1、全连接单元B2和激活单元B2;
步骤四包含如下子步骤:
步骤4.1A:将步骤三得到的高维特征图输入分支A,通过全局平均池化a1得到含有亮度信息的高维特征图,再将含有亮度信息的高维特征图通过全连接a1和激活函数a1运算,得到突出亮度信息的高维特征图,再将突出亮度信息的高维特征图通过全连接a2和激活函数a2运算,得到尺度向量sb,跳至步骤4.2;
其中,sb的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的亮度信息;
其中,全局平均池化a1的操作如式(1):
Xc表示步骤三得到的高维特征图的第c个通道,Fsq(Xc)表示含有亮度信息的高维特征图的第c个通道,H、W分别为输入的步骤三得到的高维特征图的高和宽,Xc(i,j)表示Xc第i行和第j列处的值;
步骤4.1B:将步骤三得到的高维特征图输入分支B,通过梯度运算得到梯度高维特征图,再将梯度高维特征图通过全局平均池化b1、全连接b1和激活函数b1运算,得到突出梯度信息的高维特征图,再将突出梯度信息的高维特征图通过全连接b2和激活函数b2运算,得到尺度向量sg;
其中,梯度运算在梯度运算单元中实现,全局平均池化b1在全局平均池化单元B1中实现,全连接b1使用RELU函数在全连接单元B1中实现,激活函数b1在激活单元B1中实现,全连接b2在全连接单元B2中实现,激活函数b2使用sigmoid函数在激活单元B2中实现;sg的维度记为C,与步骤三得到的高维特征图维度一致,反映了图像的细节信息;
其中,全局平均池化b1的操作如式(1),梯度运算如式(2)所示:
其中,Fgrad(Xc)表示梯度高维特征图的第c个通道,c的取值范围为1到C;Xc(i,j-1)表示Xc第i行和第j-1列处的值;Xc(i-1,j)表示Xc第i-1行和第j列处的值;
步骤4.2:将sg和sb相加,得到尺度向量s,即(3):
s=ksb+sg (3)
其中,参数k用来平衡sg和sb的权重;
步骤4.3:将步骤4.2中的尺度向量s与步骤三得到的高维特征图对应通道相乘,生成信息增强的高维特征图;
步骤五:将步骤四中得到的信息增强的高维特征图作为输入,重复Q次步骤三和步骤四,对信息增强的高维特征图进行进一步的信息提取和增强,得到信息进一步增强的高维特征图;
步骤六:将步骤五输出的信息进一步增强的高维特征图通过一个卷积层,一个批正则化层,一个激活函数运算,进行特征融合,输出融合特征图;
步骤七:将步骤六得到的融合特征图通过一个卷积核大小为1×1的卷积层,一个批正则化层,一个Tanh激活函数运算,进行降维映射,输出融合图像块;
至此,步骤三到步骤七构成了卷积神经网络F的操作;
步骤八:计算步骤七得到的融合图像块与步骤三级联图像块fp对应的红外图像块rp和可见光图像块vp之间的损失,并利用梯度下降方法回传损失,使损失函数的输出值最小化,更新卷积神经网络F中所有卷积层和全连接层的参数;
其中,损失函数如式(4)所示:
其中,L表示损失函数的输出值,If,rp,vp分别表示步骤七输出的融合图像块、组成步骤三中级联图像块的红外图像块和可见光图像块;‖·‖F表示Frobenius范数,超参数λ是调节权重的参数,取值范围在3到10之间;超参数γ是调节权重的参数,其取值范围在0.5到1.2之间;为融合图像块与红外图像块之差的Frobenius范数的平方,为融合图像块与可见光图像块之差的Frobenius范数的平方;为梯度约束条件,用于保留融合图像块的细节信息,表示梯度算子;
步骤九:判断p是否等于n,若是,则表明已得到新的卷积神经网络F中的所有卷积层和全连接层的参数,跳至步骤十;若否取p=p+1,跳至步骤三;
步骤十:固定步骤九中新的卷积神经网络F中的所有卷积层和全连接层的参数,依次将步骤一中的RTq和VTq级联得到级联图像Fq;q取1到d;
步骤十一:依次输入级联图像F1到Fd,重复步骤三到七,输出融合图像P1到Pd。
2.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤一具体为:从公开网站中下载数据集,数据集中包括m+d对红外图像和可见光图像;从数据集中随机选取m对红外图像和可见光图像作为训练集,并对红外图像编号R1到Rm,对可见光图像编号V1到Vm;剩下的d对红外图像和可见光图像作为测试集,将训练集中的所有红外图像和可见光图像进行剪裁,生成数量及大小相等的红外图像块和可见光图像块并编号。
3.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤二中p的取值范围为1到n。
4.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤三中,卷积核的大小为3×3到5×5之间,激活函数使用Leaky RELU函数,高维特征图的维度为C。
5.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤4.1A中,全局平均池化a1在全局平均池化单元A1中实现,全连接a1在全连接单元A1中实现,激活函数a1使用RELU函数在激活单元A1中实现,全连接a2在全连接单元A2中实现,激活函数a2使用sigmoid函数在激活单元A2中实现。
6.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤4.1A中,c的取值范围为1到C。
7.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤4.2中,k的取值范围在0.01到0.5之间。
8.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤五中Q的取值范围为1到5。
9.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤六中,卷积层中包括卷积核,该卷积核的大小为3×3到5×5之间。
10.如权利要求1所述的基于细节增强通道注意力的红外与可见光图像融合方法,其特征在于:步骤六中,激活函数使用Leaky RELU函数。
Applications Claiming Priority (2)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201911238636 | 2019-12-06 | ||
CN2019112386369 | 2019-12-06 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN111161201A true CN111161201A (zh) | 2020-05-15 |
CN111161201B CN111161201B (zh) | 2022-05-31 |
Family
ID=70557779
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201911338674.1A Active CN111161201B (zh) | 2019-12-06 | 2019-12-23 | 基于细节增强通道注意力的红外与可见光图像融合方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN111161201B (zh) |
Cited By (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111709903A (zh) * | 2020-05-26 | 2020-09-25 | 中国科学院长春光学精密机械与物理研究所 | 一种红外与可见光图像融合方法 |
CN113160286A (zh) * | 2021-01-06 | 2021-07-23 | 中国地质大学(武汉) | 一种基于卷积神经网络的近红外和可见光图像融合方法 |
CN113222879A (zh) * | 2021-07-08 | 2021-08-06 | 中国工程物理研究院流体物理研究所 | 一种用于红外与可见光图像融合的生成对抗网络 |
CN114782298A (zh) * | 2022-04-24 | 2022-07-22 | 西安交通大学 | 一种具有区域注意力的红外与可见光图像融合方法 |
CN115311186A (zh) * | 2022-10-09 | 2022-11-08 | 济南和普威视光电技术有限公司 | 一种红外与可见光图像跨尺度注意力对抗融合方法及终端 |
CN115423734A (zh) * | 2022-11-02 | 2022-12-02 | 国网浙江省电力有限公司金华供电公司 | 一种基于多尺度注意机制的红外与可见光图像融合方法 |
CN116664462A (zh) * | 2023-05-19 | 2023-08-29 | 兰州交通大学 | 一种基于ms-dsc和i_cbam的红外和可见光图像融合方法 |
Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US7620265B1 (en) * | 2004-04-12 | 2009-11-17 | Equinox Corporation | Color invariant image fusion of visible and thermal infrared video |
CN109447936A (zh) * | 2018-12-21 | 2019-03-08 | 江苏师范大学 | 一种红外和可见光图像融合方法 |
CN109493309A (zh) * | 2018-11-20 | 2019-03-19 | 北京航空航天大学 | 一种保持显著性信息的红外与可见光图像变分融合方法 |
CN110188685A (zh) * | 2019-05-30 | 2019-08-30 | 燕山大学 | 一种基于双注意力多尺度级联网络的目标计数方法及*** |
CN110322423A (zh) * | 2019-04-29 | 2019-10-11 | 天津大学 | 一种基于图像融合的多模态图像目标检测方法 |
-
2019
- 2019-12-23 CN CN201911338674.1A patent/CN111161201B/zh active Active
Patent Citations (5)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US7620265B1 (en) * | 2004-04-12 | 2009-11-17 | Equinox Corporation | Color invariant image fusion of visible and thermal infrared video |
CN109493309A (zh) * | 2018-11-20 | 2019-03-19 | 北京航空航天大学 | 一种保持显著性信息的红外与可见光图像变分融合方法 |
CN109447936A (zh) * | 2018-12-21 | 2019-03-08 | 江苏师范大学 | 一种红外和可见光图像融合方法 |
CN110322423A (zh) * | 2019-04-29 | 2019-10-11 | 天津大学 | 一种基于图像融合的多模态图像目标检测方法 |
CN110188685A (zh) * | 2019-05-30 | 2019-08-30 | 燕山大学 | 一种基于双注意力多尺度级联网络的目标计数方法及*** |
Non-Patent Citations (2)
Title |
---|
WANG ZHIMING,AND ETC: "Entropy based multi-resolution visible-infrared image fusion", 《2012 IEEE SYMPOSIUM ON ELECTRICAL & ELECTRONICS ENGINEERING (EEESYM)》 * |
郭玲等: "基于视觉显著性的红外与可见光图像融合", 《计算机科学》 * |
Cited By (12)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111709903A (zh) * | 2020-05-26 | 2020-09-25 | 中国科学院长春光学精密机械与物理研究所 | 一种红外与可见光图像融合方法 |
CN111709903B (zh) * | 2020-05-26 | 2022-08-19 | 中国科学院长春光学精密机械与物理研究所 | 一种红外与可见光图像融合方法 |
CN113160286A (zh) * | 2021-01-06 | 2021-07-23 | 中国地质大学(武汉) | 一种基于卷积神经网络的近红外和可见光图像融合方法 |
CN113222879A (zh) * | 2021-07-08 | 2021-08-06 | 中国工程物理研究院流体物理研究所 | 一种用于红外与可见光图像融合的生成对抗网络 |
CN113222879B (zh) * | 2021-07-08 | 2021-09-21 | 中国工程物理研究院流体物理研究所 | 一种用于红外与可见光图像融合的生成对抗网络 |
CN114782298A (zh) * | 2022-04-24 | 2022-07-22 | 西安交通大学 | 一种具有区域注意力的红外与可见光图像融合方法 |
CN114782298B (zh) * | 2022-04-24 | 2024-03-12 | 西安交通大学 | 一种具有区域注意力的红外与可见光图像融合方法 |
CN115311186A (zh) * | 2022-10-09 | 2022-11-08 | 济南和普威视光电技术有限公司 | 一种红外与可见光图像跨尺度注意力对抗融合方法及终端 |
CN115423734A (zh) * | 2022-11-02 | 2022-12-02 | 国网浙江省电力有限公司金华供电公司 | 一种基于多尺度注意机制的红外与可见光图像融合方法 |
CN115423734B (zh) * | 2022-11-02 | 2023-03-24 | 国网浙江省电力有限公司金华供电公司 | 一种基于多尺度注意机制的红外与可见光图像融合方法 |
CN116664462A (zh) * | 2023-05-19 | 2023-08-29 | 兰州交通大学 | 一种基于ms-dsc和i_cbam的红外和可见光图像融合方法 |
CN116664462B (zh) * | 2023-05-19 | 2024-01-19 | 兰州交通大学 | 一种基于ms-dsc和i_cbam的红外和可见光图像融合方法 |
Also Published As
Publication number | Publication date |
---|---|
CN111161201B (zh) | 2022-05-31 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN111161201B (zh) | 基于细节增强通道注意力的红外与可见光图像融合方法 | |
CN110298361B (zh) | 一种rgb-d图像的语义分割方法和*** | |
CN110136170B (zh) | 一种基于卷积神经网络的遥感影像建筑物变化检测方法 | |
CN111242841B (zh) | 一种基于语义分割和深度学习的图片背景风格迁移方法 | |
CN110852227A (zh) | 高光谱图像深度学习分类方法、装置、设备及存储介质 | |
CN107145836B (zh) | 基于栈式边界辨别自编码器的高光谱图像分类方法 | |
CN107506793B (zh) | 基于弱标注图像的服装识别方法及*** | |
CN109934154B (zh) | 一种遥感影像变化检测方法及检测装置 | |
CN103761295B (zh) | 基于图片自动分类的艺术类图片的定制化特征量提取方法 | |
CN109829353A (zh) | 一种基于空间约束的人脸图像风格化方法 | |
CN117033609B (zh) | 文本视觉问答方法、装置、计算机设备和存储介质 | |
CN105701493A (zh) | 基于阶层图形的图像提取以及前景估测的方法和*** | |
CN114092833A (zh) | 遥感图像分类方法、装置、计算机设备和存储介质 | |
CN112862871A (zh) | 图像融合方法及装置 | |
CN113449776A (zh) | 基于深度学习的中草药识别方法、装置及存储介质 | |
CN117036948A (zh) | 一种基于注意力机制的致敏植物识别方法 | |
CN115248876A (zh) | 基于内容理解的遥感影像统筹推荐方法 | |
CN113378812A (zh) | 一种基于Mask R-CNN和CRNN的数字表盘识别方法 | |
CN110046669B (zh) | 基于素描图像的半耦合度量鉴别字典学习的行人检索方法 | |
CN111898614B (zh) | 神经网络***以及图像信号、数据处理的方法 | |
Sjahputera et al. | Clustering of detected changes in high-resolution satellite imagery using a stabilized competitive agglomeration algorithm | |
CN114581789A (zh) | 一种高光谱图像分类方法及*** | |
CN108805280A (zh) | 一种图像检索的方法和装置 | |
CN113450297A (zh) | 红外图像和可见光图像的融合模型构建方法及*** | |
CN110222222B (zh) | 基于深层主题自编码模型的多模态检索方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |