This code is based on the paper: "Learned reconstructions for practical mask-based lensless imaging" available here: (https://www.osapublishing.org/oe/fulltext.cfm?uri=oe-27-20-28075&id=420747)
Clone this project using:
git clone https://github.com/Waller-Lab/LenslessLearning.git
The dependencies can be installed by using:
conda env create -f environment.yml
source activate lensless_learning
In addition, the LPIPS package is needed (this is used in the loss function during training). Instructions for installing LPIPS can be found here: (https://github.com/richzhang/PerceptualSimilarity)
The pre-trained models can be downloaded here
Jupyter Notebook: pre-trained reconstructions.ipynb