code and file for "C-DONN: compact diffractive optical neural network with deep learning regression"
- ComplexPyTorch folder is the Library essential for the network
- complex_fitting_50.py is the ket code for training: you can change the parameters and your training goal here
- model.py and modelpred.py is the prediction code
- matpy.m is the matlab-python interface
- test.m is the testing code
- test_data.mat is 20 samples for testing
here we offer a basic weight file in release, for further research and different mission, please train your own net by the method in the paper.