PyTorch library for solving imaging inverse problems using deep learning

Scientific computing library for optics, computer graphics and visual perception.

Modular and scalable computational imaging in Python with GPU/out-of-core computing.

(TPAMI 2025) Invertible Diffusion Models for Compressed Sensing [PyTorch]

Scientific Computational Imaging COde

Pado: Pytorch Automatic Differentiable Optics

(TPAMI 2024) Practical Compact Deep Compressed Sensing [PyTorch]

(Tensorflow Version) D-Flat is a forward and inverse design framework for flat optics. Although specially geared for the design of metasurface optics, it may be used for any end-to-end imaging and sensing task.

Image-to-image regression with uncertainty quantification in PyTorch. Take any dataset and train a model to regress images to images with rigorous, distribution-free uncertainty quantification.

Compressed Sensing: From Research to Clinical Practice with Data-Driven Learning

[CVPR'19] End-to-end Projector Photometric Compensation

Keras Implementation of the paper Residual Feature Distillation Network for Lightweight Image Super-Resolution

🚀⏱️ Official implementation for "neural space-time model for dynamic multi-shot imaging"

Clinically-Interpretable Radiomics [MICCAI'22, CMPB'21]

3D reconstructions of mm-scale objects from sequences of phone camera images

Official implementation of Neural Lithography (SIGGRAPH Asia 2023)

Repository for ptychography software

Tools for designing x-ray phantoms and experiments.

Demo for single-pixel imaging using Hadamard functions as sensing basis. Matlab/Python implementations

[ICCV'19] CompenNet++: End-to-end Full Projector Compensation