Shack-Hartmann-based wavefront and intensity sensing via U-Net

Neural networks present a new approach for solving nonlinear problems and is widely applied in optical science. In this research, we integrate neural network with Shack-Hartmann wavefront sensor (SHWS), not only reconstruct the wavefront but also the intensity of beam profile. This network has the capability to obtain the beam wavefront information without calculating the slop of wavefront, which cost most of the time in traditional algorithm, and also grab the features of beam intensity distribution simultaneously. We also compare the result of reconstruction using single focal spot. The experimental results show that though SHWS pattern training result has slightly better root mean square (RMS), both the reconstructions have high accuracy in wavefront and beam profile.