Multi-spectral and single-shot wavefront detection technique based on neural networks

Multi-spectral SHWFS spot pattern generationand neural network training programs.The Ten_sp_data.m (matlab) file generates SHWFS samples, with the target spectrum modifiable as required. The four_sp_resnet.py (python) file trains a model (using the ResNet50 architecture) on these samples to achieve synchronous measurement of multi-wavelength wavefronts.The data/label folder contains the label values for multiple sets of training and test samples. These labels can be used with the Ten_sp_data.m file to regenerate datasets or SHWFS patterns. The data/test_result folder also includes the testing set labels and the corresponding prediction errors.