CN114862976A - 一种用于旋转衍射的多光谱图像重建方法 - Google Patents

一种用于旋转衍射的多光谱图像重建方法 Download PDF

Info

Publication number
CN114862976A
CN114862976A CN202210300986.9A CN202210300986A CN114862976A CN 114862976 A CN114862976 A CN 114862976A CN 202210300986 A CN202210300986 A CN 202210300986A CN 114862976 A CN114862976 A CN 114862976A
Authority
CN
China
Prior art keywords
output
input
residual
image
hyperspectral
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.)
Pending
Application number
CN202210300986.9A
Other languages
English (en)
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.)
Zhejiang University ZJU
Zhejiang Lab
Original Assignee
Zhejiang University ZJU
Zhejiang Lab
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 Zhejiang University ZJU, Zhejiang Lab filed Critical Zhejiang University ZJU
Priority to CN202210300986.9A priority Critical patent/CN114862976A/zh
Publication of CN114862976A publication Critical patent/CN114862976A/zh
Pending legal-status Critical Current

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06TIMAGE DATA PROCESSING OR GENERATION, IN GENERAL
    • G06T11/002D [Two Dimensional] image generation
    • 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
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods

Landscapes

  • Engineering & Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Theoretical Computer Science (AREA)
  • General Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • Molecular Biology (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Artificial Intelligence (AREA)
  • Evolutionary Computation (AREA)
  • General Health & Medical Sciences (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)

Abstract

本发明公开了一种用于旋转衍射的多光谱图像重建方法。包括:获取旋转衍射多光谱图像数据集;搭建用于光谱图像重建的卷积神经网络,卷积神经网络包括编码部分、解码部分和光谱上采样残差连接部分;将旋转衍射多光谱图像数据集输入卷积神经网络并对卷积神经网络进行训练,获得训练好的卷积神经网络;将待重建旋转衍射模糊图像输入训练好的卷积神经网络中,训练好的卷积神经网络输出重建高光谱图像。本发明在重建过程中不需要点扩散函数的辅助,无需迭代求解,具有速度快,计算资源消耗少,重建结果空间分辨率和光谱精度高,输出光谱通道数量灵活等特点。

Description

一种用于旋转衍射的多光谱图像重建方法
技术领域
本发明属于光谱成像技术、深度学习领域的一种多光谱图像重建方法,具体涉及了一种用于旋转衍射的多光谱图像重建方法。
背景技术
光谱成像技术作为一种成像技术和光谱技术相结合的多维信息获取技术,与传统成像技术相比,其优势在于能探测获得被测目标的二维空间信息及一维光谱信息,是物质成分类型、热辐射特性等研究的主要信息获取手段。自上世纪光谱成像技术诞生以来,其在农业、食品、资源勘探、环境保护、生物医学、军事等领域得到了广泛的应用。
然而传统的高光谱成像系统存在着光学系统复杂、含有精密运动器件、曝光时间长等问题,极大地限制了其在许多动态场景中的应用。近年来,高光谱成像系统向着轻小型化、快照式测量方向发展,尤其是随着计算机资源的发展,出现了众多基于计算重建的轻小型快照式高光谱成像系统,旋转衍射高光谱成像系统便是其中的一种。
2019年Jeon等提出了基于旋转衍射的快照式高光谱成像系统,这种成像系统仅利用一片衍射元件辅以相应的重建算法就可实现快照式光谱成像。具有轻小型、快照式、结构简单等特点,最后实现了420nm~660nm谱段的10nm间隔,25通道高光谱成像。但是在这个系统提出的重建算法中,需要衍射元件点扩散函数的辅助,最后的输出光谱通道数必须和标定得到的点扩散函数光谱通道数保持一致,且在重建过程中存在更新迭代的步骤,计算成本较高,计算资源消耗大。
发明内容
为了解决背景技术中存在的技术问题,针对现有重建算法中存在的需要点扩散函数辅助,重建结果光谱通道数量必须与点扩散函数光谱通道数量保持一致,重建过程中需要迭代更新,计算资源消耗大等问题。本发明提出了一种用于旋转衍射的多光谱图像重建方法,在重建过程中不需要点扩散函数辅助,重建结果光谱通道数量灵活,不需要迭代更新,计算成本较低。
本发明是通过以下技术方案来实现的:
本发明包含以下步骤:
S1:获取旋转衍射多光谱图像数据集,旋转衍射多光谱图像数据集由旋转衍射模糊图像以及对应的高光谱真实图像组成;
S2:搭建用于光谱图像重建的卷积神经网络,卷积神经网络包括编码部分、解码部分和光谱上采样残差连接部分;卷积神经网络的输入分别输入到编码部分和光谱上采样残差连接部分中,编码部分的输出输入到解码部分,解码部分的输出与光谱上采样残差连接部分的输出进行按元素相加后的输出作为卷积神经网络的输出;
S3:将S1中旋转衍射多光谱图像数据集输入S2中的卷积神经网络并对卷积神经网络进行训练,获得训练好的卷积神经网络;
S4:将待重建旋转衍射模糊图像输入训练好的卷积神经网络中,训练好的卷积神经网络输出重建高光谱图像。
所述S1中旋转衍射多光谱图像数据集的获取方法包括:
利用基于位移台的旋转衍射数据集拍摄系统采集多组旋转衍射模糊图像以及对应的高光谱真实图像,从而构成旋转衍射多光谱图像数据集;
或者利用已知高光谱真实图像数据集和旋转衍射高光谱成像系统的点扩散函数计算生成旋转衍射模糊图像,由已知高光谱真实图像数据集和对应的旋转衍射模糊图像组成旋转衍射多光谱图像数据集。
所述编码部分包括五个残差卷积块和四个下采样层,编码部分的输入输入到第一残差卷积块,第一残差卷积块依次经第一下采样层、第二残差卷积块、第二下采样层、第三残差卷积块、第三下采样层、第四残差卷积块和第四下采样层后与第二残差卷积块相连,第一残差卷积块的输出作为编码部分的第一输出,第二残差卷积块的输出作为编码部分的第二输出,第三残差卷积块的输出作为编码部分的第三输出,第四残差卷积块的输出作为编码部分的第四输出,第五残差卷积块的输出作为编码部分的第五输出,编码部分的第一-五输出分别依次输入到解码部分的第一-五输入。
所述解码部分包括四个转置卷积层、四个残差卷积块和第一2D卷积层,
解码部分的第五输入输入到第一转置卷积层,第一转置卷积层的输出与解码部分的第四输入进行级联后的输出输入到第六残差卷积块,第六残差卷积块的输出与第二转置卷积层相连,第二转置卷积层的输出与解码部分的第三输入进行级联后的输出输入到第七残差卷积块,第七残差卷积块的输出与第三转置卷积层相连,第三转置卷积层的输出与解码部分的第二输入进行级联后的输出输入到第八残差卷积块,第八残差卷积块的输出与第四转置卷积层相连,第四转置卷积层的输出与解码部分的第一输入进行级联后的输出输入到第九残差卷积块,第九残差卷积块的输出与第一2D卷积层相连,第一2D卷积层的输出作为解码部分的输出。
所述光谱上采样残差连接部分包括光谱上采样残差模块,光谱上采样残差连接部分的输入和输出分别为光谱上采样残差模块的输入和输出。
所述残差卷积块包括3个2D卷积层、3个批归一化层、2个激活层和SE注意力模块,残差卷积块的输入分别输入到第三2D卷积层和第四2D卷积层中,第三2D卷积层依次经第一激活层、第一批归一化层和第五2D卷积层的输出作为注意力输入特征,注意力输入特征输入到SE注意力模块后的输出再与注意力输入特征进行按通道相乘后的输出作为中间特征,第四2D卷积层经第二批归一化层后输出与中间特征进行按元素相加后的输出输入到第二激活层,第二激活层经第三批归一化层的输出作为残差卷积块的输出。
所述S3中卷积神经网络训练时采用的总损失函数的公式为:
L=αL1+βLSSIM+γLGrad
L1=|X-Y|
Figure BDA0003562890220000031
Figure BDA0003562890220000032
其中,L为总损失函数值,L1为内容损失函数值,LSSIM为结构相似性损失函数值,LGrad为梯度差损失函数值,a、β、γ分别表示内容损失函数、结构相似性损失函数、梯度差损失函数的权重,X为重建的高光谱图像,Y为高光谱真实图像,uX、uY分别为重建的高光谱图像X、高光谱真实图像Y的均值,
Figure BDA0003562890220000033
Figure BDA0003562890220000034
分别为重建的高光谱图像X、高光谱真实图像Y的方差,σXY为重建的高光谱图像X和高光谱真实图像Y的协方差,C1和C2分别为第一、第二平滑因子,;
Figure BDA0003562890220000035
为重建的高光谱图像的梯度算子;
Figure BDA0003562890220000036
为高光谱真实图像的梯度算子,||表示取绝对值操作。
所述的旋转衍射模糊图像的生成公式如下:
Figure BDA0003562890220000037
其中,Jc(x,y)表示c颜色通道下的旋转衍射模糊图像,c表示颜色通道,满足c∈{r,g,b},r,g,b分别为红色、绿色、蓝色通道,m表示高光谱真实图像的光谱通道数量,I(x,y;λi)表示在第i个光谱通道的高光谱真实图像切片,
Figure BDA0003562890220000038
表示卷积操作,kc(x,y;λi)表示在第i个光谱通道下c颜色通道的点扩散函数。
本发明与现有技术相比,其有益效果为:
(1)本发明提出的重建方法在重建过程中不需要点扩散函数的辅助;
(2)本发明的输出高光谱图像的通道数量可以随旋转衍射数据集的通道数量灵活改变;
(3)本发明在重建过程中不需要迭代更新,计算资源消耗相对较少,网络的训练过程耗时较少。
附图说明
图1为本发明的网络结构示意图;
图2为本发明的残差卷积块示意图;
图3为生成的具有旋转衍射模糊特征的RGB图像的三个通道可视化图像;
图4为无标定旋转衍射多光谱数据集构建系统示意图;
图5为本发明的重建结果在不同谱段的图像;
图6为本发明重建结果部分色块的光谱曲线图;
图7为对比方法的重建结果在不同谱段的图像;
图8为对比方法重建结果部分色块的光谱曲线图;
图9为本发明与对比方法重建结果的空间分辨率细节对比图。
具体实施方式
下面结合附图对本发明做进一步详细的描述。
按照本发明的发明内容,完整实施的实施例及其实施过程如下:
本发明包含以下步骤:
S1:获取旋转衍射多光谱图像数据集,旋转衍射多光谱图像数据集由旋转衍射模糊图像以及对应的高光谱真实图像组成;
S1中旋转衍射多光谱图像数据集的获取方法包括:
利用基于位移台的旋转衍射数据集拍摄系统采集多组旋转衍射模糊图像以及对应的高光谱真实图像,从而构成旋转衍射多光谱图像数据集;高光谱图像为光谱分辨率高于10nm的光谱图像。
如图4所示,基于位移台的旋转衍射数据集拍摄系统包括宽谱光源1、宽谱光源2、旋转衍射高光谱成像系统3、高光谱相机4以及位移台5,宽谱光源1和宽谱光源2出射光的光谱范围至少覆盖480nm~900nm谱段。旋转衍射高光谱成像系统3与高光谱相机4设置在位移台5上,旋转衍射高光谱成像系统3与高光谱相机4的光学系统的视场角、放大率相同,且两者使用的探测器空间分辨率相同。旋转衍射高光谱成像系统3与高光谱相机4的光轴方向平行并且垂直于位移台5的移动方向。
调节位移台5使旋转衍射高光谱成像系统3的光轴正对拍摄场景,拍摄场景中的物体位于系统像面中间,拍摄一张旋转衍射模糊图像,再次调节位移台5,使得高光谱相机4的光轴正对拍摄场景,拍摄到的物体所处像面位置与旋转衍射模糊图像中物体所处像面位置保持一致,拍摄对应的高光谱真实图像。
或者利用已知高光谱真实图像数据集和旋转衍射高光谱成像系统的点扩散函数计算生成旋转衍射模糊图像,由已知高光谱真实图像数据集和对应的旋转衍射模糊图像组成旋转衍射多光谱图像数据集。
旋转衍射模糊图像的生成公式如下:
Figure BDA0003562890220000051
其中,Jc(x,y)表示c颜色通道下的旋转衍射模糊图像,c表示颜色通道,满足c∈{r,g,b},r,g,b分别为红色、绿色、蓝色通道,m表示高光谱真实图像的光谱通道数量,I(x,y;λi)表示在第i个光谱通道的高光谱真实图像切片,
Figure BDA0003562890220000052
表示卷积操作,kc(x,y;λi)表示在第i个光谱通道下c颜色通道的点扩散函数。
本实施例中,使用计算仿真的方法来构建数据对,首先对系统的点扩散函数进行标定,并对点扩散函数进行裁剪、配准、去噪,随后使用建立的61组光谱范围为480~900nm,光谱间隔10nm,43通道,分辨率2048*2048的高光谱真实图像数据集,点扩撒函数与高光谱真实图像数据集卷积生成具有旋转衍射模糊特征的旋转衍射模糊图像,由旋转衍射模糊图像与对应的高光谱真实图像组成数据对,由多组数据对构成旋转衍射多光谱数据集。从61组数据对中选取56组用于训练,剩下的5组用于验证和测试,图3的(a)、(b)、(c)所示的图像即为测试集中的RGB图像的三个通道。将分辨率为2048*2048的数据对按照相同的裁剪方法裁剪为256*256的图像块,即可用于训练神经网络。
S2:搭建用于光谱图像重建的卷积神经网络,如图1所示,卷积神经网络包括编码部分、解码部分和光谱上采样残差连接部分;卷积神经网络的输入分别输入到编码部分和光谱上采样残差连接部分中,编码部分的输出输入到解码部分,解码部分的输出与光谱上采样残差连接部分的输出进行按元素相加后的输出作为卷积神经网络的输出;
编码部分包括五个残差卷积块和四个下采样层,具体实施中,下采样层的池化函数为最大池化函数,编码部分的输入输入到第一残差卷积块,第一残差卷积块依次经第一下采样层、第二残差卷积块、第二下采样层、第三残差卷积块、第三下采样层、第四残差卷积块和第四下采样层后与第二残差卷积块相连,第一残差卷积块的输出作为编码部分的第一输出,第二残差卷积块的输出作为编码部分的第二输出,第三残差卷积块的输出作为编码部分的第三输出,第四残差卷积块的输出作为编码部分的第四输出,第五残差卷积块的输出作为编码部分的第五输出,编码部分的第一-五输出分别依次输入到解码部分的第一-五输入。
解码部分包括四个转置卷积层、四个残差卷积块和第一2D卷积层,
解码部分的第五输入输入到第一转置卷积层,第一转置卷积层的输出与解码部分的第四输入进行级联后的输出输入到第六残差卷积块,第六残差卷积块的输出与第二转置卷积层相连,第二转置卷积层的输出与解码部分的第三输入进行级联后的输出输入到第七残差卷积块,第七残差卷积块的输出与第三转置卷积层相连,第三转置卷积层的输出与解码部分的第二输入进行级联后的输出输入到第八残差卷积块,第八残差卷积块的输出与第四转置卷积层相连,第四转置卷积层的输出与解码部分的第一输入进行级联后的输出输入到第九残差卷积块,第九残差卷积块的输出与第一2D卷积层相连,第一2D卷积层的输出作为解码部分的输出,具体实施中,解码部分输出的特征为43通道。
输入图像首先经过一个残差卷积块,变为256×256×32的特征xin,随后经过第一个maxpooling下采样和残差卷积块,变为128×128×64的特征xdown1,然后经过第二个maxpooling下采样和残差卷积块,变为64×64×128的特征xdown2,随后经过第三个maxpooling下采样和残差卷积块,变为32×32×256的特征xdown3,最后经过第四个maxpooling下采样和残差卷积块,变为16×16×256的特征xdown4,至此完成编码工作。
在解码过程中,xdown4经过转置卷积上采样,与xdown3进行跳变连接,随后输入残差卷积块,得到32×32×128的特征xup1,xup1经过转置卷积上采样,与xdown2进行跳变连接,随后输入残差卷积块,得到64×64×64的特征xup2,xup2经过转置卷积上采样,与xdown1进行跳变连接,随后输入残差卷积块,得到128×128×32的特征xup3,xup3经过转置卷积上采样,与xin进行跳变连接,随后输入残差卷积块,得到256×256×64的特征xup4,xup4最后经过一个1×1大小的卷积块得到256×256×43的解码结果。解码结果与光谱上采样结果相加即可得到重建高光谱图像。
光谱上采样残差连接部分包括光谱上采样残差模块,具体实施中,光谱上采样残差模块为第二2D卷积层。光谱上采样残差连接部分的输入和输出分别为光谱上采样残差模块的输入和输出。光谱上采样残差连接部分输出尺度为256×256×43的光谱上采样结果
如图2所示,残差卷积块包括3个2D卷积层、3个批归一化层、2个激活层和SE注意力模块,具体实施中,激活层的激活函数为ReLu。残差卷积块的输入分别输入到第三2D卷积层和第四2D卷积层中,第三2D卷积层依次经第一激活层、第一批归一化层和第五2D卷积层的输出作为注意力输入特征,注意力输入特征输入到SE注意力模块后的输出再与注意力输入特征进行按通道相乘后的输出作为中间特征,第四2D卷积层经第二批归一化层后输出与中间特征进行按元素相加后的输出输入到第二激活层,第二激活层经第三批归一化层的输出作为残差卷积块的输出。
设输入的图像表示为x,则残差卷积块的作用可以按如下方式描述:
经过第一个卷积块之后的特征可以表示为x1=BN(Fact(F1(x))),其中F1(·)表示卷积核为3×3的二维卷积,Fact(·)为ReLU激活函数,BN为批量归一化操作。
随后x1经过注意力机制后的特征可以描述为x2=F2(x1)·FSE(F2(x1)),其中F2(·)表示卷积核为3×3的二维卷积,FSE(·)为SE Attention操作。SE Attention全称为Squeeze-and-Excitation Attention,是广泛使用的attention机制模块。
经过残差操作的特征可以描述为xres=BN(F3(x)),其中F3(·)表示卷积核为3×3的二维卷积。
最后残差卷积块的输入可以表示为:
xout=BN(Fact(xres+x2))
其中Fact(·)为ReLU激活函数,BN为批量归一化操作。
S3:将S1中旋转衍射多光谱图像数据集输入S2中的卷积神经网络并对卷积神经网络进行训练,获得训练好的卷积神经网络进行训练;
在训练过程中使用Adam优化器,设置学习率为10-4,每30个epoch学习率下降一半,batch size设置为8,训练结束的条件为epoch数量等于150。S3中卷积神经网络训练时采用的总损失函数的公式为:
L=aL1+βLSSIM+γLGrad
L1=|X-Y|
Figure BDA0003562890220000071
Figure BDA0003562890220000072
其中,L为总损失函数值,L1为内容损失函数值,LSSIM为结构相似性损失函数值,LGrad为梯度差损失函数值,a、β、γ分别表示内容损失函数、结构相似性损失函数、梯度差损失函数的权重,X为重建的高光谱图像,Y为高光谱真实图像,uX、uY分别为重建的高光谱图像X、高光谱真实图像Y的均值,
Figure BDA0003562890220000081
Figure BDA0003562890220000082
分别为重建的高光谱图像X、高光谱真实图像Y的方差,σXY为重建的高光谱图像X和高光谱真实图像Y的协方差,C1和C2分别为第一、第二平滑因子,是防止分母为0而设置的常数项;
Figure BDA0003562890220000083
为重建的高光谱图像的梯度算子;
Figure BDA0003562890220000084
为高光谱真实图像的梯度算子,||表示取绝对值操作。
S4:将待重建旋转衍射模糊图像输入训练好的卷积神经网络中,训练好的卷积神经网络输出重建的高光谱图像。
卷积神经网络在重建过程中不需要点扩散函数辅助,不需要迭代更新,重建速度快,计算成本低,使用1×1大小的卷积块改变光谱通道数进行光谱上采样,重建结果的光谱通道数量可以随不同旋转衍射数据集灵活改变。
选取其中一组重建结果,重建的高光谱图像在各个谱段的表现如图5所示,从图中可以看出来各个谱段的图像空间信息得到了较好的恢复。选取重建的高光谱图像中的几个色块,对光谱精度进行评价,结果如图6所示,图6的(a)为选取的重建的高光谱图像,图6的(b)-(e)分别为选取的重建的高光谱图像中第一-第四色块的光谱曲线。从图中可以看出对应色块的重建结果光谱曲线与数据集中的label保持一致,选取的色块光谱曲线与label的均方根误差(RMSE)分别为0.01531、0.02444、0.01485、0.01798。表1为本发明和其他重建方法在构建数据集上的客观评价指标对比,评价指标分别为峰值信噪比(PSNR)、结构相似性(SSIM)、光谱角制图(SAM),其中PSNR和SSIM越高越好,SAM数值越低越好。相比于其他重建方法,本发明在所有指标上都取得了较优的效果。
表1本发明和其他重建方法在构建数据集上的客观评价指标对比表
Figure BDA0003562890220000085
为了验证本发明方法的先进性,将传统的Jeon等的方法重建结果与本发明重建结果进行对比。使用峰值信噪比(PSNR)、结构相似性(SSIM)、光谱角制图(SAM)作为评价指标,其中PSNR和SSIM越高越好,SAM数值越低越好,对比结果如表1所示。从表1中可以看出来本发明在所有指标上都取得了较优的效果。除了评价指标上的对比,图7中展示了对比方法重建结果在不同谱段的表现,从图中可以看出各个谱段的复原图像清晰度不如图5,图8展示了对比方法重建结果部分色块的光谱曲线图,图8的(a)为选取的重建的高光谱图像,图8的(b)-(e)分别为选取的重建的高光谱图像中第一-第四色块的光谱曲线。从图中可以看出虽然重建结果大致正确,但是选取的色块光谱曲线与label的均方根误差(RMSE)分别为0.09163、0.07776、0.08523、0.08566,精度低于本发明的方法,图9展示了本发明与对比方法重建结果的空间分辨率细节对比图,图9的(a)和(c)分别为在500nm光谱谱段处下本发明与对比方法的重建结果,图9的(b)和(d)分别为在800nm光谱谱段处本发明与对比方法的重建结果,从图中可以看出本发明重建结果含有更丰富的空间信息。

Claims (8)

1.一种用于旋转衍射的多光谱图像重建方法,其特征在于,包含以下步骤:
S1:获取旋转衍射多光谱图像数据集,旋转衍射多光谱图像数据集由旋转衍射模糊图像以及对应的高光谱真实图像组成;
S2:搭建用于光谱图像重建的卷积神经网络,卷积神经网络包括编码部分、解码部分和光谱上采样残差连接部分;卷积神经网络的输入分别输入到编码部分和光谱上采样残差连接部分中,编码部分的输出输入到解码部分,解码部分的输出与光谱上采样残差连接部分的输出进行按元素相加后的输出作为卷积神经网络的输出;
S3:将S1中旋转衍射多光谱图像数据集输入S2中的卷积神经网络并对卷积神经网络进行训练,获得训练好的卷积神经网络;
S4:将待重建旋转衍射模糊图像输入训练好的卷积神经网络中,训练好的卷积神经网络输出重建高光谱图像。
2.根据权利要求1所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述S1中旋转衍射多光谱图像数据集的获取方法包括:
利用基于位移台的旋转衍射数据集拍摄系统采集多组旋转衍射模糊图像以及对应的高光谱真实图像,从而构成旋转衍射多光谱图像数据集;
或者利用已知高光谱真实图像数据集和旋转衍射高光谱成像系统的点扩散函数计算生成旋转衍射模糊图像,由已知高光谱真实图像数据集和对应的旋转衍射模糊图像组成旋转衍射多光谱图像数据集。
3.根据权利要求1所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述编码部分包括五个残差卷积块和四个下采样层,编码部分的输入输入到第一残差卷积块,第一残差卷积块依次经第一下采样层、第二残差卷积块、第二下采样层、第三残差卷积块、第三下采样层、第四残差卷积块和第四下采样层后与第二残差卷积块相连,第一残差卷积块的输出作为编码部分的第一输出,第二残差卷积块的输出作为编码部分的第二输出,第三残差卷积块的输出作为编码部分的第三输出,第四残差卷积块的输出作为编码部分的第四输出,第五残差卷积块的输出作为编码部分的第五输出,编码部分的第一-五输出分别依次输入到解码部分的第一-五输入。
4.根据权利要求1所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述解码部分包括四个转置卷积层、四个残差卷积块和第一2D卷积层,
解码部分的第五输入输入到第一转置卷积层,第一转置卷积层的输出与解码部分的第四输入进行级联后的输出输入到第六残差卷积块,第六残差卷积块的输出与第二转置卷积层相连,第二转置卷积层的输出与解码部分的第三输入进行级联后的输出输入到第七残差卷积块,第七残差卷积块的输出与第三转置卷积层相连,第三转置卷积层的输出与解码部分的第二输入进行级联后的输出输入到第八残差卷积块,第八残差卷积块的输出与第四转置卷积层相连,第四转置卷积层的输出与解码部分的第一输入进行级联后的输出输入到第九残差卷积块,第九残差卷积块的输出与第一2D卷积层相连,第一2D卷积层的输出作为解码部分的输出。
5.根据权利要求1所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述光谱上采样残差连接部分包括光谱上采样残差模块,光谱上采样残差连接部分的输入和输出分别为光谱上采样残差模块的输入和输出。
6.根据权利要求3或4所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述残差卷积块包括3个2D卷积层、3个批归一化层、2个激活层和SE注意力模块,残差卷积块的输入分别输入到第三2D卷积层和第四2D卷积层中,第三2D卷积层依次经第一激活层、第一批归一化层和第五2D卷积层的输出作为注意力输入特征,注意力输入特征输入到SE注意力模块后的输出再与注意力输入特征进行按通道相乘后的输出作为中间特征,第四2D卷积层经第二批归一化层后输出与中间特征进行按元素相加后的输出输入到第二激活层,第二激活层经第三批归一化层的输出作为残差卷积块的输出。
7.根据权利要求1所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述S3中卷积神经网络训练时采用的总损失函数的公式为:
L=αL1+βLSSIM+γLGrad
L1=|X-Y|
Figure FDA0003562890210000021
Figure FDA0003562890210000022
其中,L为总损失函数值,L1为内容损失函数值,LSSIM为结构相似性损失函数值,LGrad为梯度差损失函数值,α、β、γ分别表示内容损失函数、结构相似性损失函数、梯度差损失函数的权重,X为重建的高光谱图像,Y为高光谱真实图像,uX、uY分别为重建的高光谱图像X、高光谱真实图像Y的均值,
Figure FDA0003562890210000023
Figure FDA0003562890210000024
分别为重建的高光谱图像X、高光谱真实图像Y的方差,σXY为重建的高光谱图像X和高光谱真实图像Y的协方差,C1和C2分别为第一、第二平滑因子,;
Figure FDA0003562890210000031
为重建的高光谱图像的梯度算子;
Figure FDA0003562890210000032
为高光谱真实图像的梯度算子,||表示取绝对值操作。
8.根据权利要求2所述的一种用于旋转衍射的多光谱图像重建方法,其特征在于,所述的旋转衍射模糊图像的生成公式如下:
Figure FDA0003562890210000033
其中,Jc(x,y)表示c颜色通道下的旋转衍射模糊图像,c表示颜色通道,满足c∈{r,g,b},r,g,b分别为红色、绿色、蓝色通道,m表示高光谱真实图像的光谱通道数量,I(x,y;λi)表示在第i个光谱通道的高光谱真实图像切片,
Figure FDA0003562890210000034
表示卷积操作,kc(x,y;λi)表示在第i个光谱通道下c颜色通道的点扩散函数。
CN202210300986.9A 2022-03-24 2022-03-24 一种用于旋转衍射的多光谱图像重建方法 Pending CN114862976A (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202210300986.9A CN114862976A (zh) 2022-03-24 2022-03-24 一种用于旋转衍射的多光谱图像重建方法

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202210300986.9A CN114862976A (zh) 2022-03-24 2022-03-24 一种用于旋转衍射的多光谱图像重建方法

Publications (1)

Publication Number Publication Date
CN114862976A true CN114862976A (zh) 2022-08-05

Family

ID=82628933

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202210300986.9A Pending CN114862976A (zh) 2022-03-24 2022-03-24 一种用于旋转衍射的多光谱图像重建方法

Country Status (1)

Country Link
CN (1) CN114862976A (zh)

Cited By (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN117058038A (zh) * 2023-08-28 2023-11-14 北京航空航天大学 一种基于偶数卷积深度学习的衍射模糊图像修复方法

Cited By (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN117058038A (zh) * 2023-08-28 2023-11-14 北京航空航天大学 一种基于偶数卷积深度学习的衍射模糊图像修复方法
CN117058038B (zh) * 2023-08-28 2024-04-30 北京航空航天大学 一种基于偶数卷积深度学习的衍射模糊图像修复方法

Similar Documents

Publication Publication Date Title
CN111260576B (zh) 一种基于去噪三维卷积自编码网络的高光谱解混算法
CN110501072B (zh) 一种基于张量低秩约束的快照式光谱成像系统的重构方法
CN113222823B (zh) 基于混合注意力网络融合的高光谱图像超分辨率方法
CN109064396A (zh) 一种基于深度成分学习网络的单幅图像超分辨率重建方法
CN111080567A (zh) 基于多尺度动态卷积神经网络的遥感图像融合方法及系统
CN114972803B (zh) 一种基于联合优化的快照式光谱成像方法和系统
CN111709882B (zh) 基于亚像素卷积与特征分割的超分辨率融合的计算方法
Zeng et al. Robust reconstruction with deep learning to handle model mismatch in lensless imaging
CN107560735B (zh) 基于压缩感知的高分辨率红外成像系统及方法
CN113008370B (zh) 基于液晶高光谱计算成像系统的三维自适应压缩重构方法
Liu et al. An efficient residual learning neural network for hyperspectral image superresolution
CN113744136A (zh) 基于通道约束多特征融合的图像超分辨率重建方法和系统
CN114862976A (zh) 一种用于旋转衍射的多光谱图像重建方法
CN115272093A (zh) 一种基于空间结构信息约束的高光谱图像解混方法
CN116403046A (zh) 一种高光谱影像分类装置及方法
Deng et al. Multiple frame splicing and degradation learning for hyperspectral imagery super-resolution
CN112784747B (zh) 高光谱遥感图像多尺度本征分解方法
CN116773018A (zh) 一种计算光谱成像的空谱联合图像重建方法及系统
CN108801457B (zh) 基于编码采样板设计以及二次能量约束校正的三维图谱获取及重建方法
Li et al. Fused recurrent network via channel attention for remote sensing satellite image super-resolution
CN116612009A (zh) 一种多尺度连接生成对抗网络医学图像超分辨率重建方法
CN116433548A (zh) 一种基于多级信息提取的高光谱与全色图像融合方法
Zhu et al. Super-resolution of sentinel-2 images based on deep channel-attention residual network
CN114791320A (zh) 折衍射混合的两次对焦图像融合多光谱成像系统和方法
CN113899453A (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