This is the Git repository for the CNN model of the paper Deep learning-driven wavefront sensing for grating-array based wavefront sensor.
The program was developed in Python 3.8.8 with Tensorflow 2.10 (CPU). Recently it was testd on Python 3.10.12 with Tensorflow 2.17.1 (on Google colab)
I would suggest you use Anaconda / Miniconda and set up a virtual environment as:
conda create -n "myenv" python=3.8.8# replace "myenv" with your desired name.
After setting up the virtual environment, you can install the dependencies as:
pip install -r requirements.txt
Alternatively, you can use Google Colab to process your code.
- training_data_generator.ipynb : this notebook is used to generate training as well as testing data for the CNN model
- CNN_GAWS_v1.ipynb : this is the main notebook which predicts the aberration strengths from focal spots array
A step-by-step procedure for using each notebook is incorporated into the notebooks.
Thank you,
Pranjal Choudhury