├── .gitignore ├── README.md ├── Reconstruction_demo.ipynb ├── assets └── Intro.png ├── helper ├── __pycache__ │ ├── data_setup.cpython-311.pyc │ ├── data_setup.cpython-37.pyc │ ├── data_setup.cpython-39.pyc │ ├── engine.cpython-311.pyc │ ├── engine.cpython-37.pyc │ ├── engine.cpython-39.pyc │ ├── logging_metric.cpython-311.pyc │ ├── logging_metric.cpython-37.pyc │ ├── logging_metric.cpython-39.pyc │ ├── model_builder.cpython-311.pyc │ ├── model_builder.cpython-37.pyc │ └── model_builder.cpython-39.pyc ├── data_setup.py ├── engine.py ├── logging_metric.py ├── model_builder.py └── scheduler.py ├── helper_functions.py ├── models ├── __pycache__ │ ├── admm_helper.cpython-311.pyc │ ├── admm_helper.cpython-37.pyc │ ├── admm_helper.cpython-39.pyc │ ├── admm_model.cpython-311.pyc │ ├── admm_model.cpython-37.pyc │ ├── admm_model.cpython-39.pyc │ ├── admm_rgb.cpython-311.pyc │ ├── admm_rgb.cpython-37.pyc │ ├── admm_rgb.cpython-39.pyc │ ├── ensemble.cpython-311.pyc │ ├── ensemble.cpython-39.pyc │ ├── network_swinir.cpython-311.pyc │ ├── network_swinir.cpython-37.pyc │ ├── network_swinir.cpython-39.pyc │ ├── u_net2.cpython-311.pyc │ ├── u_net2.cpython-39.pyc │ └── unet.cpython-311.pyc ├── admm_helper.py ├── admm_model.py ├── admm_rgb.py ├── ensemble.py ├── network_swinir.py └── u_net2.py ├── requirements.txt ├── result_test.py ├── sample_images ├── diffuser │ ├── im157.npy │ ├── im189.npy │ ├── im326.npy │ ├── im394.npy │ ├── im408.npy │ ├── im43.npy │ ├── im643.npy │ ├── im65.npy │ └── im740.npy ├── lensed │ ├── im157.npy │ ├── im189.npy │ ├── im326.npy │ ├── im394.npy │ ├── im408.npy │ ├── im43.npy │ ├── im643.npy │ ├── im65.npy │ └── im740.npy └── psf.tiff ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | dataset/ 2 | checkpoint.pth 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/README.md -------------------------------------------------------------------------------- /Reconstruction_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/Reconstruction_demo.ipynb -------------------------------------------------------------------------------- /assets/Intro.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/assets/Intro.png -------------------------------------------------------------------------------- /helper/__pycache__/data_setup.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/data_setup.cpython-311.pyc -------------------------------------------------------------------------------- /helper/__pycache__/data_setup.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/data_setup.cpython-37.pyc -------------------------------------------------------------------------------- /helper/__pycache__/data_setup.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/data_setup.cpython-39.pyc -------------------------------------------------------------------------------- /helper/__pycache__/engine.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/engine.cpython-311.pyc -------------------------------------------------------------------------------- /helper/__pycache__/engine.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/engine.cpython-37.pyc -------------------------------------------------------------------------------- /helper/__pycache__/engine.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/engine.cpython-39.pyc -------------------------------------------------------------------------------- /helper/__pycache__/logging_metric.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/logging_metric.cpython-311.pyc -------------------------------------------------------------------------------- /helper/__pycache__/logging_metric.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/logging_metric.cpython-37.pyc -------------------------------------------------------------------------------- /helper/__pycache__/logging_metric.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/logging_metric.cpython-39.pyc -------------------------------------------------------------------------------- /helper/__pycache__/model_builder.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/model_builder.cpython-311.pyc -------------------------------------------------------------------------------- /helper/__pycache__/model_builder.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/model_builder.cpython-37.pyc -------------------------------------------------------------------------------- /helper/__pycache__/model_builder.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/__pycache__/model_builder.cpython-39.pyc -------------------------------------------------------------------------------- /helper/data_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/data_setup.py -------------------------------------------------------------------------------- /helper/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/engine.py -------------------------------------------------------------------------------- /helper/logging_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/logging_metric.py -------------------------------------------------------------------------------- /helper/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/model_builder.py -------------------------------------------------------------------------------- /helper/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper/scheduler.py -------------------------------------------------------------------------------- /helper_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/helper_functions.py -------------------------------------------------------------------------------- /models/__pycache__/admm_helper.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_helper.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_helper.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_helper.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_helper.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_helper.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_model.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_model.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_model.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_model.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_model.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_model.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_rgb.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_rgb.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_rgb.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_rgb.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/admm_rgb.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/admm_rgb.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/ensemble.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/ensemble.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/ensemble.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/ensemble.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/network_swinir.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/network_swinir.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/network_swinir.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/network_swinir.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/network_swinir.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/network_swinir.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/u_net2.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/u_net2.cpython-311.pyc -------------------------------------------------------------------------------- /models/__pycache__/u_net2.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/u_net2.cpython-39.pyc -------------------------------------------------------------------------------- /models/__pycache__/unet.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/__pycache__/unet.cpython-311.pyc -------------------------------------------------------------------------------- /models/admm_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/admm_helper.py -------------------------------------------------------------------------------- /models/admm_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/admm_model.py -------------------------------------------------------------------------------- /models/admm_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/admm_rgb.py -------------------------------------------------------------------------------- /models/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/ensemble.py -------------------------------------------------------------------------------- /models/network_swinir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/network_swinir.py -------------------------------------------------------------------------------- /models/u_net2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/models/u_net2.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/requirements.txt -------------------------------------------------------------------------------- /result_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/result_test.py -------------------------------------------------------------------------------- /sample_images/diffuser/im157.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im157.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im189.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im189.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im326.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im326.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im394.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im394.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im408.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im408.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im43.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im43.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im643.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im643.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im65.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im65.npy -------------------------------------------------------------------------------- /sample_images/diffuser/im740.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/diffuser/im740.npy -------------------------------------------------------------------------------- /sample_images/lensed/im157.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im157.npy -------------------------------------------------------------------------------- /sample_images/lensed/im189.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im189.npy -------------------------------------------------------------------------------- /sample_images/lensed/im326.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im326.npy -------------------------------------------------------------------------------- /sample_images/lensed/im394.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im394.npy -------------------------------------------------------------------------------- /sample_images/lensed/im408.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im408.npy -------------------------------------------------------------------------------- /sample_images/lensed/im43.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im43.npy -------------------------------------------------------------------------------- /sample_images/lensed/im643.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im643.npy -------------------------------------------------------------------------------- /sample_images/lensed/im65.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im65.npy -------------------------------------------------------------------------------- /sample_images/lensed/im740.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/lensed/im740.npy -------------------------------------------------------------------------------- /sample_images/psf.tiff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/sample_images/psf.tiff -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/arpanpoudel/lenslessimaging/HEAD/utils.py --------------------------------------------------------------------------------