一种基于图像融合和卷积神经网络的相位恢复方法  

Phase retrieval wavefront sensing based on image fusion and convolutional neural network

作　　者：周静 张晓芳[1] 赵延庚 Zhou Jing;Zhang Xiao-Fang;Zhao Yan-Geng(School of Optics and Photonics,Beijing Institute of Technology,Beijing 100081,China)

机构地区：[1]北京理工大学光电学院,北京100081

出　　处：《物理学报》2021年第5期124-132,共9页Acta Physica Sinica

基　　金：国家自然科学基金(批准号:61471039)资助的课题.

摘　　要：相位恢复法利用光波传输中某一(或某些)截面上的光强分布来传感系统波前,其结构简单,不易受震动及环境干扰,被广泛应用于光学遥感和像差检测等领域.传统相位恢复法采用迭代计算,很难满足实时性要求,且在一定程度上依赖于迭代转换或迭代优化初值.为克服上述问题,本文提出了一种基于卷积神经网络的相位恢复方法,该方法采用基于小波变换的图像融合技术对焦面和离焦面图像进行融合处理,可在不损失图像信息的同时简化卷积神经网络的输入.网络模型训练完成后可依据输入的融合图像直接输出表征波前相位的4-9阶Zernike系数,且波前传感精度均方根(root-mean-square,RMS)可达0.015λ,λ=632.8 nm.研究了噪声、离焦量误差和图像采样分辨率等因素对波前传感精度的影响,验证了该方法对噪声具有一定鲁棒性,相对离焦量误差在7.5%内时,波前传感精度RMS仍可达0.05λ,且随着图像采样分辨率的提升,波前传感精度有所改善,但训练时间成本随之增加.此外,分析了实际应用中,当系统像差阶数与网络训练阶数略有差异时,本方法所能实现的传感精度,并给出了解决方案.The conventional phase retrieval wavefront sensing approaches mainly refer to a series of iterative algorithms,such as G-S algorithms,Y-G algorithms and error reduction algorithms.These methods use intensity information to calculate the wavefront phase.However,most of the traditional phase retrieval algorithms are difficult to meet the real-time requirements and depend on the iteration initial value used in iterative transformation or iterative optimization to some extent,so their practicalities are limited.To solve these problems,in this paper,a phase-diversity phase retrieval wavefront sensing method based on wavelet transform image fusion and convolutional neural network is proposed.Specifically,the image fusion method based on wavelet transform is used to fuse the point spread functions at the in-focus and defocus image planes,thereby simplifying the network inputs without losing the image information.The convolutional neural network(CNN) can directly extract image features and fit the required nonlinear mapping.In this paper,the CNN is utilized to establish the nonlinear mapping between the fusion images and wavefront distortions(represented by Zernike polynomials),that is,the fusion images are taken as the input data,and the corresponding Zernike coefficients as the output data.The network structure of the training in this paper has 22 layers,they are 1 input layer,13 convolution layers,6 pooling layers,1 flatten layer and 1 full connection layer,that is,the output layer.The size of the convolution kernel is 3 × 3 and the step size is 1.The pooling method selects the maximum pooling and the size of the pooling kernel is 2 × 2.The activation function is ReLU,the optimization function is Adam,the loss function is the MSE,and the learning rate is 0.0001.The number of training data is 10000,which is divided into three parts:training set,validation set,and test set,accounting for 80%,15% and 5% respectively.Trained CNN can directly output the Zernike coefficients of order 4-9 to a high precision,with these fus

关 键 词：相位恢复 卷积神经网络 图像融合 传感精度 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TP183[自动化与计算机技术—计算机科学与技术]