PtychoNN is a two-headed encoder-decoder network that simultaneously predicts sample amplitude and phase from input diffraction data alone. PtychoNN is 100s of times faster than iterative phase retrieval and can work with as little as 5X less data.
Companion repository to the paper at: https://aip.scitation.org/doi/full/10.1063/5.0013065
The strucuture of the network is shown below:
git lfs
-- Data to train the network is hosted at the git lfs server due to its size
Tensorflow 1.14
Keras 2.2.4
Tf2 folder contains notebooks compatible with TF 2.x
-- NOTE: notebooks were run in Google Colab, modify as required for local runtimes