一种四台阶相位型菲涅尔波带片的快速波前复原方法

1.本发明涉及无波前传感器的单帧图像波前复原的技术领域，尤其涉及一种四台阶相位型菲涅尔波带片的快速波前复原方法。


背景技术：

2.相位反演方法是一种利用焦面光强分布来解算波前的探测方法，该方法结构简单，可实现高阶连续像差检测，能对扩展目标进行相位复原，可在线检测，能量集中度高，适合弱光探测。但其主要问题是反演精度不高，且存在多解问题，在迭代过程容易陷入停滞。这种迭代停滞是由于实际的近场复振幅f(x,y)和它的旋转180度共轭复振幅在远场上具有相同的光强分布而产生的。因此，gs算法很容易收敛到局部最小伪解或两个全局最小模糊解之一。
3.单帧焦面图像的波前反演方法是通过一张远场图像实现波前复原，该方法光路简单，但是多解现象的存在使其应用范围收到了限制。法国的serge meimon等人，通过在原像差中加上一定角度的散光，得到低阶像差与光斑分布之间的线性算法，但此种方法的线性区域受到限制，只对低阶像差探测有效(参见c.plantet,s.meimon,j.m.conan,et al.experiment al validation of lift for estimation of low-order modes in low-flux wavefront sensing[j].optics express,2013,21(14):16337-16352)。dong bing等人在2015年提出了一种混合相位反演算法，该算法使用了线性相位反演和gs算法相结合。利用线性反演的结果作为gs算法的初始波前相位。通过此方法，高阶像差基本恢复，但多解问题仍未解决(参见b.dong,g.jia,r.wang.hybrid phase retrieval from a single defocused image[c].proc.spie,2015,9622:962205)。
[0004]
基于单帧焦面图像的波前相位反演方法，具有结构简单、环境要求低、能量利用率高的优点，是一种具有广阔应用前景和研究价值的波前复原技术。但传统的单帧焦面反演技术存在多解问题，会导致测试效果不够精确。相位差法虽然可以克服此问题，但它需要在焦面和离焦面两个位置探测光强信息，结构较复杂。其他的基于单帧焦面图像的波前反演方法存在各种限制，并没有根本解决相位反演的复原精度问题。
[0005]
因此，本发明提出一种基于混合焦距的四台阶相位型菲涅尔波带片结合神经网络的波前复原法。


技术实现要素：

[0006]
本发明要解决的技术问题是：在只采集一张远场光斑图像的情况下，使远场既包含一部分在焦信息，又包含一部分离焦信息，然后结合神经网络利用远场图像反演近场波前相位唯一解，并保证波前复原的精度和实时性。本发明提供一种基于混合焦距的四台阶相位型菲涅尔波带片结合神经网络的波前复原法，设计一个一、三象限为一焦距，二、四象限为另一焦距的四台阶菲涅尔波带片，打破近场波前的旋转翻转对称性，从根本上克服多解问题，另外结合非线性拟合能力强大的神经网络技术，无需耗时量大的迭代寻优即可得
到最终复原结果，实现一种更为实时高效的无波前探测自适应光学波前复原方法。
[0007]
本发明解决上述问题采用的技术方案是：一种四台阶相位型菲涅尔波带片的快速波前复原方法，选用在样本空间充分采样的数据集供卷积神经网络(cnn)拟合远场光斑与近场波前相位的映射关系，网络训练收敛之后，输入经混合焦距菲涅尔波带片成像的远场光斑图像，即可获得其对应的波前泽尼克模式系数，此映射求解过程不再需要迭代运算，减少计算耗时，具体实现步骤如下：
[0008]
步骤1：设计基于混合焦距的四台阶相位型菲涅尔波带片，包括混合焦距的分布、入瞳半径、焦距差、相位台阶宽度、相位台阶高度、衍射效率参数设计；
[0009]
步骤2：验证步骤1设计的基于混合焦距的四台阶相位型菲涅尔波带片是否可以在混合焦距的四台阶相位型菲涅尔波带片的其中之一个焦距焦面上同时获得单帧点光源的远场光强分布，并且光强利用率达到百分之八十以上，且光斑半径大于五个像素，否则神经网络对少量的像素数难以进行特征提取；
[0010]
步骤3：若步骤2可以实现，则使用步骤1设计的混合焦距的四台阶相位型菲涅尔波带片，对理想平行光引入符合大气传输模型的随机像差，记录远场光斑与近场波前数据，并将ccd探测到的远场光斑图像和与之一一对应的的近场波前zernike模式系数(或近场波面)分别作为样本和标签制作基于混合焦距的四台阶相位型菲涅尔波带片结合神经网络的波前复原数据集，若步骤2无法实现，则重复执行步骤1再次设计基于混合焦距的四台阶相位型菲涅尔波带片，直至实现步骤2；
[0011]
步骤4：配置深度学习环境，搭建神经网络；
[0012]
步骤5：任意抽取数据集中80％的样本作为训练集，供网络学习远场光斑与近场波前的映射关系；从数据集剩下的20％样本中任意抽取一半样本做为验证集，用于调整网络超参数并验证算法有效性，最后剩余的10％样本作为测试集用于进行无偏估计；
[0013]
步骤6：更换不同的神经网络进行测试，记录测试集结果，根据最终结果选出性能最优的网络结构。
[0014]
其中，所述步骤1中混合焦距的四台阶相位型菲涅尔波带片的第一、三象限为一个焦距值，第二、四象限为与一、三象限不同的焦距值，将ccd放置在一、三或二、四象限的在焦面上才可保证远场成像既包含一部分在焦信息又包含一部分离焦信息，达到打破近场波面翻转旋转对称性的目的，即克服多解问题。
[0015]
其中，所述步骤1中混合焦距的四台阶相位型菲涅尔波带片中一、三象限与二、四象限的焦距值差值可根据传统相位差法确认一个合适的离焦量，更有利于提高波前复原的精度，优选一个波长左右的离焦量。
[0016]
其中，所述步骤2中选择衍射效率较高且台阶数适中的的四台阶类型进行设计，台阶宽度、高度直接影响衍射效率。
[0017]
其中，所述步骤3中数据集的样本数应至少为三万组，即应在样本空间进行充分采样。
[0018]
其中，所述步骤5中训练集、验证集与测试集的划分方式可根据实际需求和数据集的大小适当改变。
[0019]
本发明与现有技术相比的优点在于：
[0020]
(1)本发明设计的混合焦距的四台阶相位型菲涅尔波带片衍射效率较高且仅需一
张远场光斑图像即可克服多解问题；
[0021]
(2)本发明仿真设计了多种离焦量和多种相位台阶高度的四台阶菲涅尔波带片，使用神经网络训练测试确定了复原精度较高的设计参数，复原62阶泽尼克模式系数的精度可达波前复原残差为0.008个波长；
[0022]
(3)本发明无传统的波前传感器和额外调制器件，结构简单易操作，成像设备仅需一台ccd相机；
[0023]
(4)本发明神经网络引入了通道注意力和空间注意力机制，让网络更多的聚焦于重要局部信息，抑制无用信息，无需迭代，实时性强，并且更换多种网络测试网络性能选出波前复原效果最优的网络结构，预测一帧波面推理时间小于4ms。
附图说明
[0024]
图1为本发明一种四台阶相位型菲涅尔波带片的快速波前复原方法的控制算法工作原理示意图；
[0025]
图2为本发明一种四台阶相位型菲涅尔波带片的快速波前复原方法的工作流程图；
[0026]
图3为基于混合焦距的四台阶相位型菲涅尔波带片的无波前传感原理图；
[0027]
图4为本发明采用的神经网络架构示意图，其中，图4(a)为神经网络整体结构，图4(b)为通道注意力模块，图4(c)为空间注意力模块。
具体实施方式
[0028]
为使本发明的目的、技术方案和优点更加清楚明白，以下结合具体实施案例，并参照附图，对本发明进一步详细说明。
[0029]
图1是一种四台阶相位型菲涅尔波带片的快速波前复原方法原理示意图。图2是本发明一种四台阶相位型菲涅尔波带片的快速波前复原方法的工作流程图，具体的实施过程为：
[0030]
步骤1：设计混合焦距的四台阶相位型菲涅尔波带片，将ccd置于混合焦距的四台阶相位型菲涅尔波带片的一、三象限或二、四象限焦距之一的焦平面。混合焦距的四台阶相位型菲涅尔波带片相关参数如下：中心波长为520nm，入瞳直径12mm，一、三象限焦距为50cm，二、四象限焦距为52cm，台阶数为4，等间距分布，一三象限相邻弧带相位分布依次为0,0.5*pi,pi,1.5*pi，二四象限相邻弧带相位分布依次为0,0.65*pi,1.3*pi,1.95*pi。图3为混合焦距的四台阶相位型菲涅尔波带片示意图，图3为基于混合焦距的四台阶相位型菲涅尔波带片的无波前传感原理图；
[0031]
步骤2：验证步骤1设计的菲涅尔波带片是否可以分别针对正、负离焦近场波前光束在经菲涅尔波带片在一、三象限或二、四象限焦距之一的焦平面获得不同形态的远场光强分布；
[0032]
步骤3：若步骤2可以实现，则根据步骤1收集经混合焦距的四台阶相位型菲涅尔波带片成像的远场光斑与近场波前数据，并将ccd放置在一、三象限或二、四象限焦距之一的焦平面，焦面远场图像和近场波前对应的zernike模式系数分别作为数据集中的样本和标签，若步骤2无法实现，则重复执行步骤1再次设计传感器，直至实现步骤2；
[0033]
步骤4：配置深度学习环境，搭建cnn，图4为本发明的注意力机制cnn架构示意图：注意力机制cnn架构总共包含11层，其中有1层输入层，4层卷积层，4层池化层，2层全连接层并且加入通道注意力机制和空间注意力机制。输入层输入224
×
224大小的远场图像样本。4层卷积层的卷积核大小分别为5
×
5、5
×
5、4
×
4、4
×
4，4层卷积层的通道数分别为16、16、32、32。池化层均选择最大池化，池化层步长均为2。两层全连接层的节点数分别为200和62。网络输出4到65阶zernike系数。cnn选择adam函数作为梯度下降函数，初始学习率为10-3
。为了避免网络过拟合，对每层卷积层都引入batch正则化。epoch设置为1000，batchsize大小设置为64。
[0034]
步骤5：任意抽取数据集中80％的样本作为训练集，供网络学习远场光斑与近场波前的映射关系；从数据集剩下的20％样本中任意抽取一半样本做为验证集，用于调整网络超参数并验证算法有效性，最后剩余的10％样本作为测试集用于进行无偏估计。
[0035]
训练使网络收敛后，只需给网络输入单帧焦面光强图像，网络即可输出该远场光斑对应的近场波前信息。此过程中不再涉及迭代运算，计算速度大大提高，仿真结果表明，本发明完成一次波前复原所需时间可小于4毫秒。
[0036]
步骤1中，设定混合焦距的四台阶相位型菲涅尔波带片第一象限和第三象限的焦距为f
13
，第二和第四象限焦距为f
24
，按照波带片理论，第n个环带半径可以表示为：
[0037][0038][0039]
式中，λ为光学波长，r
13_n
表示第一象限和第三象限的第n个环带半径，r
24_n
表示第二和第四象限的第n个环带半径。混合焦距的四台阶相位型菲涅尔波带片的相邻四个环带具有相位差，第一象限和第三象限的调制相位可以表示为：
[0040][0041]
第二象限和第四象限的调制相位可以表示为：
[0042][0043]
因此，混合焦距的四台阶相位型菲涅尔波带片后表面的波前复振幅表示为：
[0044]
[0045]
式中，u
near
(x1，y1，0)为混合焦距的四台阶相位型菲涅尔波带片前表面的近场光波复振幅。
[0046]
根据菲涅耳衍射基本理论，当ccd相机靶面位于混合焦距的四台阶相位型菲涅尔波带片后f
13
的位置时，远场光斑复振幅可以表示为：
[0047][0048]
式中(x2+y2)为远场空间二维坐标，λ为波长，k＝2π/λ为波数，表示傅里叶变换。
[0049]
步骤4中，注意力机制的加入目的是当给定一个中间特征图，让网络沿着通道和空间两个维度依次推断出注意力权重，然后与原特征图相乘来对特征进行自适应调整。注意力机制模块是一个轻量级的通用模块，它可以无缝地集成到任何cnn架构中，额外开销忽略不计，并且可以与基本cnn一起进行端到端的训练。
[0050]
通道注意力机制是关注什么样的特征是有意义的。为了汇总空间特征，采用了全局平均池化和最大池化两种方式来分别利用不同的信息：
[0051][0052]
输入是一个h
×w×
c的特征f(h
×
w代表像素大小)，先分别进行一个空间的全局最大池化和平均池化得到两个1
×1×
c的通道描述。接着，再将它们分别送入一个两层的神经网络，第一层神经元个数为c/r，激活函数为relu，第二层神经元个数为c，这个两层的神经网络是共享的。然后再将得到的两个特征相加后经过一个sigmoid激活函数得到权重系数mc。最后，输入的特征f与mc相乘即可得到缩放后的新特征。
[0053]
在通道注意力模块之后，再引入空间注意力模块来关注哪里的特征是有意义的：
[0054][0055]
与通道注意力相似，给定一个h
×w×
c的特征f，先分别进行一个通道维度的最大池化和平均池化得到两个h
×w×
i的通道描述，并将这两个描述按照通道拼接在一起。然后，经过一个7
×
7的卷积层，激活函数为sigmoid，得到权重系数ms。最后，输入的特征f与ms相乘即可得到缩放后的新特征。
[0056]
通道注意力和空间注意力这两个模块可以以并行或者顺序的方式组合在一起，本发明发现顺序组合并且将通道注意力放在前面可以取得更好的效果，因此采用先通道注意力后空间注意力的串联的方式分别加入到第二层卷积层和最后一层卷积层后面。
[0057]
以上所述，仅为本发明中的具体实施方式，但本发明的保护范围并不局限于此，任何熟悉该技术的人在本发明所揭露的技术范围内，可理解想到的变换或替换，都应涵盖在本发明的包含范围之内。