CN113034408B - 一种红外热成像深度学习图像去噪方法及装置 - Google Patents

一种红外热成像深度学习图像去噪方法及装置 Download PDF

Info

Publication number
CN113034408B
CN113034408B CN202110482836.XA CN202110482836A CN113034408B CN 113034408 B CN113034408 B CN 113034408B CN 202110482836 A CN202110482836 A CN 202110482836A CN 113034408 B CN113034408 B CN 113034408B
Authority
CN
China
Prior art keywords
feature map
module
image
layer
convolution
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.)
Active
Application number
CN202110482836.XA
Other languages
English (en)
Other versions
CN113034408A (zh
Inventor
程良伦
李卓
吴衡
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Guangdong University of Technology
Original Assignee
Guangdong University of Technology
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by Guangdong University of Technology filed Critical Guangdong University of Technology
Priority to CN202110482836.XA priority Critical patent/CN113034408B/zh
Publication of CN113034408A publication Critical patent/CN113034408A/zh
Application granted granted Critical
Publication of CN113034408B publication Critical patent/CN113034408B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T5/00Image enhancement or restoration
    • G06T5/70Denoising; Smoothing
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T7/00Image analysis
    • G06T7/10Segmentation; Edge detection
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20021Dividing image into blocks, subimages or windows
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T2207/00Indexing scheme for image analysis or image enhancement
    • G06T2207/20Special algorithmic details
    • G06T2207/20081Training; Learning

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • General Physics & Mathematics (AREA)
  • General Health & Medical Sciences (AREA)
  • General Engineering & Computer Science (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Data Mining & Analysis (AREA)
  • Evolutionary Computation (AREA)
  • Artificial Intelligence (AREA)
  • Molecular Biology (AREA)
  • Computing Systems (AREA)
  • Biomedical Technology (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Mathematical Physics (AREA)
  • Software Systems (AREA)
  • Health & Medical Sciences (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Image Analysis (AREA)
  • Image Processing (AREA)

Abstract

本发明公开了一种红外热成像系统深度学习图像去噪方法及装置,所述方法包括:获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;将输入图像作为训练好的深度学习模型中,输出噪声的估计值;从所述目标物图像中去除噪声的估计值,得到去噪后的图像。本方法,可以充分利用图像丰富的结构特征和相似性,更好地利用图像块特征中的上下文信息并充分提取图像特征,在红外图像本身特征值较少情况下实现了较好的效果,非常有利于红外热成像及深度学习技术的应用研究。

Description

一种红外热成像深度学习图像去噪方法及装置
技术领域
本发明涉及热红外成像领域,具体涉及一种红外热成像系统深度学习图像去噪方法及装置。
背景技术
由于黑体辐射的存在,任何物体都依据温度的不同对外进行电磁波辐射。波长为2.0~1000微米的部分称为热红外线。热红外成像通过对热红外敏感对物体进行成像,能反映出物体表面的温度场。热红外在军事、工业、汽车辅助驾驶、医学领域都有广泛的应用。
近年来,随着红外热成像技术广泛应用,进一步拓宽了人类视觉的可视范围,但也暴露出一些问题,例如红外图像成像质量差、噪声干扰严重,尤其是对于温度平均的环境下,由于缺少热源,其成像效果会进一步受损。
发明内容
本发明的目的是提供一种红外热成像系统深度学习图像去噪方法及装置,用以解决现有红外成像系统存在的成像质量差、噪声干扰严重的问题。
为了实现上述任务,本发明采用以下技术方案:
本申请的第一方面提供了一种红外热成像系统深度学习图像去噪方法,包括以下步骤:
获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;
对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;
构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;
将输入图像作为训练好的深度学习模型中,输出噪声的估计值;
从所述目标物图像中去除噪声的估计值,得到去噪后的图像。
进一步地,所述深度学习模型包括依次连接的第一卷积模块、第一区域级非局部模块、残差密集块、第二区域级非局部模块、第二卷积模块、上采样模块以及重建模块,其中:
所述第一卷积模块用于对输入图像进行浅层特征提取,得到第一特征图;第一特征图一方面作为所述第一区域级非局部模块的输入,另一方面与所述第二卷积模块的输出叠加后,作为上采样模块的输入;
所述第一区域级非局部模块用于对第一特征图进行非局部增强处理,得到第二特征图;
所述残差密集块用于对于所述第二特征图进行深层特征的提取,以生成深层特征图;
所述第二区域级非局部模块用于对深层特征图进行非局部增强处理,得到第三特征图;
所述第二卷积模块用于对第三特征图进行特征提取的同时调整通道,之后与所述第一特征图叠加后,生成第四特征图;
所述上采样模块用于对第四特征图尺寸恢复;
所述重建模块用于对尺寸恢复后的第四特征图进行通道调整,重建为噪声残差。
进一步地,所述第一区域级非局部模块、第二区域级非局部模块的结构相同,包括四个卷积层χ,φ,κ和γ;
第一特征图首先被划分为k×k个具有相同大小的特征图块,特征图块分别经过χ,φ,κ进行卷积操作,生成特征图
Figure BDA0003049894780000021
Figure BDA0003049894780000022
Figure BDA0003049894780000023
Figure BDA0003049894780000024
进行乘法操作,生成特征图
Figure BDA0003049894780000025
Figure BDA0003049894780000026
对特征图中特征位置索引总和进行除法操作来完成归一化,之后与
Figure BDA0003049894780000027
再次进行乘法操作以生成特征图
Figure BDA0003049894780000028
之后经过卷积层γ处理之后,
Figure BDA0003049894780000029
生成特征图
Figure BDA00030498947800000210
最后,
Figure BDA00030498947800000211
与特征图Ft 1叠加完成非局部增强,得到多个增强特征图像块Ft R1;多个增强特征图像块合并得到第二特征图FR1
进一步地,所述残差密集块包括多个基础块,每个基础块由多个残差模块和一个二阶信道注意力模块SOCA组成,每个残差模块所提取的特征图信息由缩放系数β控制;
残差模块包括多层卷积层和带泄露修正线性整流层LReLU的连续堆叠和跳转连接,之后为一个卷积层;
所述二阶信道注意力模块SOCA包括依次连接的协方差归一化层、全局协方差池化层、卷积层、线性整流层,卷积层,Sigmoid层。
进一步地,在残差密集块中,第二特征图经过残差模块中的多层卷积层和带泄露修正线性整流层LReLU的处理之后,被传入二阶信道注意力模块SOCA处理,通过利用二阶特征统计量自适应地学习特征间的依赖关系,来重新调整通道特征。
进一步地,在二阶信道注意力模块SOCA中,协方差归一化层对处理后的第二特征图进行处理,将特征图转化为归一化协方差矩阵,接着传入全局协方差池化层进行收缩,得到信道方向的统计量,统计量再经过卷积层、线性整流层,卷积层,Sigmoid层处理后,生成信道关注图,通过信道关注图重新分配所述的第二特征图。
进一步地,第二特征图完成二阶信道注意力模块SOCA的处理后,残差密集块中的第一个基础块中,第一个残差模块生成特征图F1 d,它经过缩放系数β的缩放,与第二特征图FR1叠加传入第二残差模块;在经过与第一个残差模块相同的处理之后,生成的特征图
Figure BDA0003049894780000031
经过缩放与F1 d叠加传入到第三个残差模块;第三残差模块对特征图进行同样的处理,生成特征图
Figure BDA0003049894780000032
Figure BDA0003049894780000033
经过缩放后与
Figure BDA0003049894780000034
叠加,再次进行缩放并与特征图FR1叠加生成特征图F1 D,完成第一个基础块的处理;随后,处理后的特征图被传入后续的其他基础块进行相同的处理,完成对特征的深度提取,生成深层特征图
Figure BDA0003049894780000035
进一步地,所述重建模块包括一系列依次连接的卷积层,在最后一个卷积层之前的每个卷积层之后均有一个线性整流层ReLU;
第四特征图传入重建模块后通过卷积层和线性整流层ReLU调整通道,最终通过最后一层卷积层重建为噪声残差。
一种红外热成像系统深度学习图像去噪装置,包括:
分割单元,用于获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;
预处理单元,用于对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;
加噪单元,用于构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;
神经网络单元,用于将输入图像作为训练好的深度学习模型中,输出噪声的估计值;
去噪单元,用于从所述目标物图像中去除噪声的估计值,得到去噪后的图像。
第三方面,本申请提供了一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,处理器执行计算机程序时实现前述第一方面的红外热成像系统深度学习图像去噪方法的步骤。
第四方面,本申请提供了一种计算机可读存储介质,计算机可读存储介质存储有计算机程序,计算机程序被处理器执行时实现前述第一方面的红外热成像系统深度学习图像去噪方法的步骤。
与现有技术相比,本发明具有以下技术特点:
本发明的图像去噪方法,通过采用“下采样分割”技术,提高了速度,降低复杂度,并且可通过调节参数来应对不同的目标图像大小;通过“区域级非局部模块RL-NL”,可以充分利用图像丰富的结构特征和相似性,更好地利用图像块特征中的上下文信息。相比如现有技术,如FFDnet去噪算法,本发明采用数据集混合训练方式,提高了算法的泛化能力,在针对不同的环境都有较好的去噪效果;对比CBDnet去噪算法,本发明采用二阶信道注意力机制SOCA,充分提取图像特征,在红外图像本身特征值较少情况下实现了较好的效果,本发明非常有利于红外热成像及深度学习技术的应用研究。
附图说明
图1为本发明一个实施例提供的图像去噪方法的流程示意图;
图2为实施例中经预处理所得到的图像,该示例为将大小为1×320×240的图像处理为5×160×120,噪声图M作为额外通道将添加到输入中;
图3为实施例中深度学习模型的架构示意图;
图4为实施例中区域级非局部模块示意图;
图5为实施例中残差密集块内部基本块示意图;
图6为实施例中残差块内部示意图,该示例中β=0.2;
图7为实施例中二阶信道注意力机制SOCA示意图;
图8为实施例中重建模块示意图。
具体实施方式
参见图1,本申请的第一方面提供了一种红外热成像系统深度学习图像去噪方法,包括以下步骤:
S1,获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像。
本申请的一个实施例中,利用红外热成像系统拍摄被测目标物图像I,其大小为nch×h×w,对目标物图像I进行一次下采样分割成4nch×h/2×w/2的子图像。其中nch为信道,h为高度,w为宽度。
S2,对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像。
该实施例中,对每个图像进行2×2图像块的提取,并在输出图像的不同通道中重组其像素,得到预处理后的子图像A0,其数学模型表示如下:
Figure BDA0003049894780000051
上述表达式中,c为图像信道,x为图像像素横坐标,y为图像像素纵坐标,其中0≤c≤4nch,0≤x≤h,0≤y≤w。大多数处理将在以上缩小的规模下进行。
S3,构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像。
从[σ12)均匀分布中随机采样,输出nch×h×w大小的样本来构建噪声图M。噪声图M将作为额外通道添加到输入图像A0中。该噪声图控制了降噪和细节保留之间的权衡。
经上述步骤预处理,得到的目标图像大小为(4nch+nch)×h/2×w/2的图像A1,其示意图如图2所示。
S4,将输入图像作为训练好的深度学习模型中,输出噪声的估计值。
如图3所示,将经预处理后的图像A1输入到深度学习模型,得到噪声的估计值,其数学模型可以表示为:
N=F(A1)
上述表达式中,N为噪声预测值,F(·)为一个隐函数,表示深度神经网络。
S5,从所述目标物图像中去除噪声的估计值,得到去噪后的图像。
经过深度学习模型处理之后,从目标图像I中去除预测噪声N,即I-N输出即为去噪后的图像IDN,其数学模型可以表示如下:
IDN=I-N
上式中IDN表示去噪后的图像,I为目标图像,N是深度学习模型预测噪声。
作为上述技术方案的一种可实现的方式,所述深度学习模型包括依次连接的第一卷积模块、第一区域级非局部模块、残差密集块、第二区域级非局部模块、第二卷积模块、上采样模块以及重建模块,其中:
所述第一卷积模块用于对输入图像进行浅层特征提取,得到第一特征图;第一特征图一方面作为所述第一区域级非局部模块的输入,另一方面与所述第二卷积模块的输出叠加后,作为上采样模块的输入;所述第一区域级非局部模块用于对第一特征图进行非局部增强处理,得到第二特征图;所述残差密集块用于对于所述第二特征图进行深层特征的提取,以生成深层特征图;所述第二区域级非局部模块用于对深层特征图进行非局部增强处理,得到第三特征图;所述第二卷积模块用于对第三特征图进行特征提取的同时调整通道,之后与所述第一特征图叠加后,生成第四特征图;所述上采样模块用于对第四特征图尺寸恢复;所述重建模块用于对尺寸恢复后的第四特征图进行通道调整,重建为噪声残差。
本方案的深度学习模型结构,示例性地,如图3所示;本实施例中,从左到右,该深度学习模型依次包括:第一卷积模块、第一区域级非局部模块RN-NL、残差密集块、第二区域级非局部模块RN-NL、第二卷积模块、上采样模块以及重建模块,各模块介绍如下:
1.第一卷积模块
第一卷积模块为一个3×3卷积层,输入图像A1首先经过3×3卷积层,在该层中对输入图像A1进行浅层特征提取,得到第一特征图F1,第一特征图一方面进入到后续分支进行处理,另一方面通过跳转线传入到深层网络,与所述第二卷积模块的输出叠加后,作为上采样模块的输入;即为网络模型深层提供浅层特征并与深层特征图叠加后,传入上采样模块进行处理。
2.第一区域级非局部模块RL-NL
本方案提供了一种RL-NL模块结构,如图4所示,RL-NL模块中包括4个1×1卷积层,其中左侧3个卷积层从上至下分别被标注为χ,φ,和κ,右侧卷积层被标注为γ;在RL-NL模块中,输入的大小为h×w的特征图F1首先被划分成k×k个具有相同大小的特征图块,其划分过程的可以表示如下:
h1=h/k,w1=w/k
Figure BDA0003049894780000071
上述表达式中,h1和w1分别为被划分后的图像块大小,Ft 1代表第t个特征图像块,每个特征图像块均进行非局部增强;输入的特征图块Ft 1首先通过χ,φ,和κ卷积操作,生成特征图
Figure BDA0003049894780000072
Figure BDA0003049894780000073
Figure BDA0003049894780000074
Figure BDA0003049894780000075
进行乘法操作,生成特征图
Figure BDA0003049894780000076
Figure BDA0003049894780000077
对特征图中特征位置索引总和n=h×w/k2进行除法操作来完成归一化,之后与
Figure BDA0003049894780000078
再次进行乘法操作以生成特征图
Figure BDA0003049894780000079
在经过1×1卷积层γ处理之后,
Figure BDA00030498947800000710
生成特征图
Figure BDA00030498947800000711
最后,
Figure BDA00030498947800000712
与特征图Ft 1叠加完成非局部增强。上述非局部增强的数学模型可表示为:
Figure BDA00030498947800000713
其中,i是要计算的特征图的特征位置索引,j是特征图中所有可能位置的索引,
Figure BDA00030498947800000714
表示第t个特征图像块的第i个位置,Ft R1表示增强后Ft 1
Figure BDA00030498947800000715
分别表示1×1卷积层χ,φ和κ的卷积处理,Wχ,Wφ,Wκ,和Wγ为1×1卷积层χ,φ,κ和γ学习到的权重,N=h×w/k2为特征图的位置索引个数。
最后,增强特征图像块Ft R1合并生成第二特征图FR1,第一区域级非局部模块处理完成。
3.残差密集块
本方案的一个示例中,残差密集块由5个基础块组成,如图5所示,基础块由3个残差模块组成,每个残差模块所提取的特征图信息被缩放系数β控制。残差模块的网络结构如图6所示,从左到右,该网络包括4层3×3卷积层和带泄露修正线性整流层LReLU的连续堆叠和跳转连接,1层3×3卷积层,二阶信道注意力模块SOCA。图7展示了二阶信道注意力模块SOCA的网络结构,包括协方差归一化层,全局协方差池化层,3×3卷积层,线性整流层ReLU,3×3卷积层,Sigmoid层。
在残差密集块中,增强后的第二特征图FR1被输入到第一个基础块进行深度特征提取,第二特征图FR1经过残差模块中一系列3×3卷积层和带泄露修正线性整流层LReLU的处理之后,被传入二阶信道注意力模块SOCA处理,通过利用二阶特征统计量自适应地学习特征间的依赖关系,来重新调整通道特征。在二阶信道注意力模块SOCA中,协方差归一化层对前面处理后的第二特征图FR1进行处理,将特征图转化为归一化协方差矩阵,其过程如下:
首先,大小为C×H×W的特征图FR1的数学模型被表示为FR1=[f1,…,fC],并被重塑为特征矩阵X,则其样本协方差矩阵Σ可表示为:
Figure BDA0003049894780000081
其中
Figure BDA0003049894780000082
s=WH为信道所在维度特征,es和I分别是s×s单位矩阵和所有都为1的矩阵。由于协方差矩阵Σ对称半正定,因此进行特征值分解:
Σ=UΛUT
在表达式中,U为正交矩阵,UT为U的转置,Λ为特征值非递增顺序的对角矩阵。之后,将矩阵转化为特征值的幂:
Σα=UΛαUT
其中α=0.5,
Figure BDA0003049894780000083
经过协方差归一化层的处理,第二特征图FR1生成归一化协方差矩阵,接着传入全局协方差池化层进行收缩,得到信道方向的统计量z∈RC×1。本方案令
Figure BDA0003049894780000084
则统计量z的c维度的数学模型可以表示为:
Figure BDA0003049894780000085
其中,yc为信道描述符的一个分量,C为信道描述符分量总数,zc为该分量上的统计量。
为了通过全局协方差池充分利用聚集信息的特征相关性,特征图所形成的信道统计量z被传入之后的3×3卷积层,线性整流层ReLU,3×3卷积层,Sigmoid层以生成信道关注图w,其数学模型可表示为:
w=Sigmoid(Wgδ(Wdz))
上述表达式中,Wg和Wd分别为两个3×3卷积层的权重集,Sigmoid(·)和δ(·)是sigmoid函数和ReLu函数。通过w重新分配输入的特征图FR1=[f1,…,fC],其数学表达式为:
Figure BDA0003049894780000091
其中,wc和fc表示第c个通道中的比例因子和特征图。
第二特征图完成二阶信道注意力模块SOCA的通道调整后,第一个基础块中的第一个残差模块生成特征图F1 d,它经过缩放系数β的缩放,与特征图FR1叠加传入第二个残差模块。在经过与第一个残差模块相同的处理之后,生成的特征图
Figure BDA0003049894780000092
经过缩放与F1 d叠加传入到第三个残差模块。第三个残差模块对特征图进行同样的处理,生成特征图
Figure BDA0003049894780000093
Figure BDA0003049894780000094
经过缩放后与
Figure BDA0003049894780000095
叠加,再次进行缩放并与特征图FR1叠加生成特征图F1 D,完成第一个基础块的处理。随后,特征图被传入第二、三、四、五个基础块进行相同的处理,完成对特征的深度提取,生成深层特征图
Figure BDA0003049894780000096
4.第二区域级非局部模块
本方案中的第二RN-NL模块与第一RN-NL模块具有相同的结构,并对特征图进行相同的非局部增强处理以生成第三特征图FR2
5.第二卷积模块
第二卷积模块为一个3×3卷积层,深层特征图FR2经过3×3卷积层处理,进行特征提取同时调整通道;之后,特征图与浅层网络传来的第一特征图F1叠加生成第四特征图F′。
6.上采样模块
在上采样模块中,将尺寸为h/2×w/2的第四特征图重新恢复为h×w尺寸,生成特征图FU,该处理通过反转下采样公式来实现,其中x为图像长度,y为图像宽度,c为图像通道数。
Figure BDA0003049894780000097
7.重建模块
重建模块由一系列3×3卷积层和线性整流层ReLU组成,如图8所示,从左至右依次为为:3×3卷积层,线性整流层ReLU,3×3卷积层,线性整流层ReLU,3×3卷积层。特征图FU传入重建模块后被调整通道,最终由一层3×3卷积层重建为nch×h×w大小的噪声残差。
本方案的深度学习模型的训练中,采用Adam函数对损失函数L(Θ)进行优化,其过程表示如下:
Figure BDA0003049894780000101
上式中
Figure BDA0003049894780000102
为经预处理后的图像块,Mj为每次预处理所加入的噪声估计图,Θ为深度学习网络参数,
Figure BDA0003049894780000103
为输入图像中噪声的估计值,m输入图像块数目。
其中
Figure BDA0003049894780000104
Figure BDA0003049894780000105
是由训练集中
Figure BDA0003049894780000106
中的
Figure BDA0003049894780000107
的经上述预处理得到,训练集
Figure BDA0003049894780000108
可分为三种:
(1)真实噪声训练集(训练集中的
Figure BDA0003049894780000109
为高清图像,
Figure BDA00030498947800001010
为含有真实噪声的图像)
(2)加性噪声训练集(训练集中的
Figure BDA00030498947800001011
为高清图像,
Figure BDA00030498947800001012
为含有加性高斯白噪声的图像)
(3)合训练集,将(1),(2)两种训练集都进行训练。
经过2000次(可以通过程序控制)训练后,可以得到优化参数Θ。
对于一个由红外热成像系统拍摄的目标物图像I,并经过图像预处理得到的子图像A1,可以通过IDN=I-F(A1)进行去噪从而得到清晰的图像IDN
本发明方法提高了成像质量并且有效去除图像中的噪声干扰,特别是在应对图像中目标物体温度比较平均的时,即无热源,图像出现很多噪点的情况下有着出色表现,比如,应对阴暗面拍摄的红外图像,其有望在生物医学、遥感探测、汽车辅助驾驶等领域得到广泛应用。
根据本申请的另一方面,提供了一种红外热成像系统深度学习图像去噪装置,包括:
分割单元,用于获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;
预处理单元,用于对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;
加噪单元,用于构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;
神经网络单元,用于将输入图像作为训练好的深度学习模型中,输出噪声的估计值;
去噪单元,用于从所述目标物图像中去除噪声的估计值,得到去噪后的图像。
需要说明的是,上述各个模块的具体功能和相关解释参见前述方法实施例中对应的步骤,在此不赘述。
本申请实施例进一步提供一种终端设备,该终端设备可以为计算机、服务器;包括存储器、处理器以及存储在存储器中并可在处理器上运行的计算机程序,处理器执行计算机程序时实现上述红外热成像系统深度学习图像去噪方法的步骤,例如,前述的S1至S5。
计算机程序也可以被分割成一个或多个模块/单元,一个或者多个模块/单元被存储在存储器中,并由处理器执行,以完成本申请。一个或多个模块/单元可以是能够完成特定功能的一系列计算机程序指令段,该指令段用于描述计算机程序在终端设备中的执行过程,例如,计算机程序可以被分割为分割单元、预处理单元、加噪单元、神经网络单元以及去噪单元,各模块的功能参见前述装置中的描述,不再赘述。
本申请的实施提供了一种计算机可读存储介质,计算机可读存储介质存储有计算机程序,计算机程序被处理器执行时实现上述红外热成像系统深度学习图像去噪方法的步骤,例如,前述的S1至S5。
集成的模块/单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请实现上述实施例方法中的全部或部分流程,也可以通过计算机程序来指令相关的硬件来完成,计算机程序可存储于一计算机可读存储介质中,该计算机程序在被处理器执行时,可实现上述各个方法实施例的步骤。其中,计算机程序包括计算机程序代码,计算机程序代码可以为源代码形式、对象代码形式、可执行文件或某些中间形式等。计算机可读介质可以包括:能够携带计算机程序代码的任何实体或装置、记录介质、U盘、移动硬盘、磁碟、光盘、计算机存储器、只读存储器(ROM,Read-Only Memory)、随机存取存储器(RAM,Random Access Memory)、电载波信号、电信信号以及软件分发介质等。需要说明的是,计算机可读介质包含的内容可以根据司法管辖区内立法和专利实践的要求进行适当的增减,例如在某些司法管辖区,根据立法和专利实践,计算机可读介质不包括电载波信号和电信信号。
以上实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围,均应包含在本申请的保护范围之内。

Claims (8)

1.一种红外热成像系统深度学习图像去噪方法,其特征在于,包括以下步骤:
获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;
对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;
构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;
将输入图像作为训练好的深度学习模型中,输出噪声的估计值;
从所述目标物图像中去除噪声的估计值,得到去噪后的图像;
所述深度学习模型包括依次连接的第一卷积模块、第一区域级非局部模块、残差密集块、第二区域级非局部模块、第二卷积模块、上采样模块以及重建模块,其中:
所述第一卷积模块用于对输入图像进行浅层特征提取,得到第一特征图;第一特征图一方面作为所述第一区域级非局部模块的输入,另一方面与所述第二卷积模块的输出叠加后,作为上采样模块的输入;
所述第一区域级非局部模块用于对第一特征图进行非局部增强处理,得到第二特征图;
所述残差密集块用于对于所述第二特征图进行深层特征的提取,以生成深层特征图;
所述第二区域级非局部模块用于对深层特征图进行非局部增强处理,得到第三特征图;
所述第二卷积模块用于对第三特征图进行特征提取的同时调整通道,之后与所述第一特征图叠加后,生成第四特征图;
所述上采样模块用于对第四特征图尺寸恢复;
所述重建模块用于对尺寸恢复后的第四特征图进行通道调整,重建为噪声残差;
所述第一区域级非局部模块、第二区域级非局部模块的结构相同,包括四个卷积层χ,φ,κ和γ;
第一特征图首先被划分为k×k个具有相同大小的特征图块,特征图块分别经过χ,φ,κ进行卷积操作,生成特征图
Figure FDA0003674438150000021
Figure FDA0003674438150000022
Figure FDA0003674438150000023
Figure FDA0003674438150000024
进行乘法操作,生成特征图
Figure FDA0003674438150000025
Figure FDA0003674438150000026
对特征图中特征位置索引总和进行除法操作来完成归一化,之后与
Figure FDA0003674438150000027
再次进行乘法操作以生成特征图
Figure FDA0003674438150000028
之后经过卷积层γ处理之后,
Figure FDA0003674438150000029
生成特征图
Figure FDA00036744381500000210
最后,
Figure FDA00036744381500000211
与特征图Ft 1叠加完成非局部增强,得到多个增强特征图像块Ft R1;多个增强特征图像块合并得到第二特征图FR1
2.根据权利要求1所述的红外热成像系统深度学习图像去噪方法,其特征在于,所述残差密集块包括多个基础块,基础块由多个残差模块和一个二阶信道注意力模块SOCA组成,每个残差模块所提取的特征图信息由缩放系数β控制;
残差模块包括多层卷积层和带泄露修正线性整流层LReLU的连续堆叠和跳转连接,之后为一个卷积层;
所述二阶信道注意力模块SOCA包括依次连接的协方差归一化层、全局协方差池化层、卷积层、线性整流层,卷积层,Sigmoid层。
3.根据权利要求2所述的红外热成像系统深度学习图像去噪方法,其特征在于,在残差密集块中,第二特征图经过残差模块中的多层卷积层和带泄露修正线性整流层LReLU的处理之后,被传入二阶信道注意力模块SOCA处理,通过利用二阶特征统计量自适应地学习特征间的依赖关系,来重新调整通道特征。
4.根据权利要求2所述的红外热成像系统深度学习图像去噪方法,其特征在于,在二阶信道注意力模块SOCA中,协方差归一化层对处理后的第二特征图进行处理,将特征图转化为归一化协方差矩阵,接着传入全局协方差池化层进行收缩,得到信道方向的统计量,统计量再经过卷积层、线性整流层,卷积层,Sigmoid层处理后,生成信道关注图,通过信道关注图重新分配所述的第二特征图。
5.根据权利要求2所述的红外热成像系统深度学习图像去噪方法,其特征在于,第二特征图完成二阶信道注意力模块SOCA的处理后,残差密集块的第一个基础块中,第一个残差模块生成特征图F1 d,它经过缩放系数β的缩放,与第二特征图FR1叠加传入第二残差模块;在经过与第一个残差模块相同的处理之后,生成的特征图
Figure FDA0003674438150000031
经过缩放与F1 d叠加传入到第三个残差模块;第三残差模块对特征图进行同样的处理,生成特征图
Figure FDA0003674438150000032
Figure FDA0003674438150000033
经过缩放后与
Figure FDA0003674438150000034
叠加,再次进行缩放并与特征图FR1叠加生成特征图F1 D,完成第一个基础块的处理;随后,处理后的特征图被传入后续的其他基础块进行相同的处理,完成对特征的深度提取,生成深层特征图
Figure FDA0003674438150000035
6.根据权利要求1所述的红外热成像系统深度学习图像去噪方法,其特征在于,所述重建模块包括一系列依次连接的卷积层,在最后一个卷积层之前的每个卷积层之后均有一个线性整流层ReLU;
第四特征图传入重建模块后通过卷积层和线性整流层ReLU调整通道,最终通过最后一层卷积层重建为噪声残差。
7.一种红外热成像系统深度学习图像去噪装置,其特征在于,包括:
分割单元,用于获取红外热成像系统采集的目标物图像并进行下采样处理,分割成子图像;
预处理单元,用于对每个子图像进行图像块提取以及重组像素,得到预处理后的子图像;
加噪单元,用于构建噪声图,并将噪声图作为额外通道添加到所述预处理后的子图像中作为输入图像;
神经网络单元,用于将输入图像作为训练好的深度学习模型中,输出噪声的估计值;
去噪单元,用于从所述目标物图像中去除噪声的估计值,得到去噪后的图像;
所述深度学习模型包括依次连接的第一卷积模块、第一区域级非局部模块、残差密集块、第二区域级非局部模块、第二卷积模块、上采样模块以及重建模块,其中:
所述第一卷积模块用于对输入图像进行浅层特征提取,得到第一特征图;第一特征图一方面作为所述第一区域级非局部模块的输入,另一方面与所述第二卷积模块的输出叠加后,作为上采样模块的输入;
所述第一区域级非局部模块用于对第一特征图进行非局部增强处理,得到第二特征图;
所述残差密集块用于对于所述第二特征图进行深层特征的提取,以生成深层特征图;
所述第二区域级非局部模块用于对深层特征图进行非局部增强处理,得到第三特征图;
所述第二卷积模块用于对第三特征图进行特征提取的同时调整通道,之后与所述第一特征图叠加后,生成第四特征图;
所述上采样模块用于对第四特征图尺寸恢复;
所述重建模块用于对尺寸恢复后的第四特征图进行通道调整,重建为噪声残差;
所述第一区域级非局部模块、第二区域级非局部模块的结构相同,包括四个卷积层χ,φ,κ和γ;
第一特征图首先被划分为k×k个具有相同大小的特征图块,特征图块分别经过χ,φ,κ进行卷积操作,生成特征图
Figure FDA0003674438150000041
Figure FDA0003674438150000042
Figure FDA0003674438150000043
Figure FDA0003674438150000044
进行乘法操作,生成特征图
Figure FDA0003674438150000045
Figure FDA0003674438150000046
对特征图中特征位置索引总和进行除法操作来完成归一化,之后与
Figure FDA0003674438150000047
再次进行乘法操作以生成特征图
Figure FDA0003674438150000048
之后经过卷积层γ处理之后,
Figure FDA0003674438150000049
生成特征图
Figure FDA00036744381500000410
最后,
Figure FDA00036744381500000411
与特征图Ft 1叠加完成非局部增强,得到多个增强特征图像块Ft R1;多个增强特征图像块合并得到第二特征图FR1
8.一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,处理器执行计算机程序时实现根据权利要求1至6中任一权利要求所述方法的步骤。
CN202110482836.XA 2021-04-30 2021-04-30 一种红外热成像深度学习图像去噪方法及装置 Active CN113034408B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202110482836.XA CN113034408B (zh) 2021-04-30 2021-04-30 一种红外热成像深度学习图像去噪方法及装置

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202110482836.XA CN113034408B (zh) 2021-04-30 2021-04-30 一种红外热成像深度学习图像去噪方法及装置

Publications (2)

Publication Number Publication Date
CN113034408A CN113034408A (zh) 2021-06-25
CN113034408B true CN113034408B (zh) 2022-08-12

Family

ID=76455335

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202110482836.XA Active CN113034408B (zh) 2021-04-30 2021-04-30 一种红外热成像深度学习图像去噪方法及装置

Country Status (1)

Country Link
CN (1) CN113034408B (zh)

Families Citing this family (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113469910B (zh) * 2021-06-29 2023-03-24 展讯通信(上海)有限公司 图像处理方法、装置及设备
CN113487530B (zh) * 2021-08-02 2023-06-16 广东工业大学 一种基于深度学习的红外与可见光融合成像方法

Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN109285129A (zh) * 2018-09-06 2019-01-29 哈尔滨工业大学 基于卷积神经网络的图像真实噪声去除系统
CN109583344A (zh) * 2018-11-21 2019-04-05 红相股份有限公司 一种基于深度学习的电力设备红外图像处理方法
CN109949235A (zh) * 2019-02-26 2019-06-28 浙江工业大学 一种基于深度卷积神经网络的胸部x光片去噪方法
CN111476732A (zh) * 2020-04-03 2020-07-31 江苏宇特光电科技股份有限公司 一种图像融合及去噪的方法及系统
CN112085655A (zh) * 2020-08-21 2020-12-15 东南大学 一种基于密集残差注意面部先验网络的人脸超分辨率方法
CN112200750A (zh) * 2020-10-21 2021-01-08 华中科技大学 一种超声图像去噪模型建立方法及超声图像去噪方法
CN112703509A (zh) * 2018-08-07 2021-04-23 布赖凯科技股份有限公司 用于图像增强的人工智能技术

Family Cites Families (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113362223B (zh) * 2021-05-25 2022-06-24 重庆邮电大学 基于注意力机制和双通道网络的图像超分辨率重建方法

Patent Citations (7)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN112703509A (zh) * 2018-08-07 2021-04-23 布赖凯科技股份有限公司 用于图像增强的人工智能技术
CN109285129A (zh) * 2018-09-06 2019-01-29 哈尔滨工业大学 基于卷积神经网络的图像真实噪声去除系统
CN109583344A (zh) * 2018-11-21 2019-04-05 红相股份有限公司 一种基于深度学习的电力设备红外图像处理方法
CN109949235A (zh) * 2019-02-26 2019-06-28 浙江工业大学 一种基于深度卷积神经网络的胸部x光片去噪方法
CN111476732A (zh) * 2020-04-03 2020-07-31 江苏宇特光电科技股份有限公司 一种图像融合及去噪的方法及系统
CN112085655A (zh) * 2020-08-21 2020-12-15 东南大学 一种基于密集残差注意面部先验网络的人脸超分辨率方法
CN112200750A (zh) * 2020-10-21 2021-01-08 华中科技大学 一种超声图像去噪模型建立方法及超声图像去噪方法

Non-Patent Citations (1)

* Cited by examiner, † Cited by third party
Title
基于去模糊空间变换RCNN的毫米波图像目标检测;程良伦 等;《光子学报》;20200229;第49卷(第2期);第1-2页 *

Also Published As

Publication number Publication date
CN113034408A (zh) 2021-06-25

Similar Documents

Publication Publication Date Title
CN112233026A (zh) 一种基于多尺度残差注意力网络的sar图像去噪方法
CN112819910B (zh) 基于双鬼注意力机制网络的高光谱图像重建方法
CN111583123A (zh) 一种基于小波变换的融合高低频信息的图像增强算法
CN113034408B (zh) 一种红外热成像深度学习图像去噪方法及装置
CN116797488A (zh) 一种基于特征融合与注意力嵌入的低照度图像增强方法
CN110363068B (zh) 一种基于多尺度循环生成式对抗网络的高分辨行人图像生成方法
Kim et al. Deeply aggregated alternating minimization for image restoration
Min et al. Blind deblurring via a novel recursive deep CNN improved by wavelet transform
CN112085717B (zh) 一种用于腹腔镜手术的视频预测方法及其系统
CN116091916A (zh) 一种多尺度从rgb图像重建对应的高光谱图像算法及系统
CN115578262A (zh) 基于afan模型的偏振图像超分辨率重建方法
CN114830168A (zh) 图像重建方法、电子设备和计算机可读存储介质
CN116188274A (zh) 一种图像超分辨率重建方法
Huang et al. Binarizing super-resolution networks by pixel-correlation knowledge distillation
CN113487530B (zh) 一种基于深度学习的红外与可见光融合成像方法
CN113762277B (zh) 一种基于Cascade-GAN的多波段红外图像融合方法
CN114936977A (zh) 一种基于通道注意力和跨尺度特征融合的图像去模糊方法
Dharejo et al. SwinWave-SR: Multi-scale lightweight underwater image super-resolution
CN117593187A (zh) 基于元学习和Transformer的遥感图像任意尺度超分辨率重建方法
CN117593275A (zh) 一种医学图像分割系统
Liu et al. Dnt: Learning unsupervised denoising transformer from single noisy image
CN116266336A (zh) 视频超分辨率重建方法、装置、计算设备及存储介质
CN112598581B (zh) 一种rdn超分辨网络的训练方法及图像生成方法
CN113674149A (zh) 一种基于卷积神经网络的新型超分辨率重建方法
CN113902647A (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