File size: 1,477 Bytes
41bbf3e 1dde2c6 41bbf3e 1dde2c6 5be9343 1dde2c6 03821d4 1dde2c6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | ---
library_name: transformers
tags:
- image-restoration
- lensless-imaging
license: mit
---
# Lensless Computational Imaging
10 models for reconstructing images from a lensless camera.
Trained on [DigiCam-Mirflickr-MultiMask-10K](https://huggingface.co/datasets/bezzam/DigiCam-Mirflickr-MultiMask-10K).
## Models
**Classical**
| Subfolder | Description |
|---|---|
| `admm100` | ADMM, 100 iterations, fixed hyperparameters. |
| `fista100` | FISTA, 100 iterations, fixed hyperparameters. |
**Unrolled**
| Subfolder | Description |
|---|---|
| `le_admm20` | ADMM, 20 unrolled steps with learnable per-step hyperparameters. |
| `le_fista20` | FISTA, 20 unrolled steps with learnable per-step hyperparameters. |
**Modular**
| Subfolder | Description |
|---|---|
| `modular_le_admm5_prepost` | Pre-UNet → 5 ADMM steps → post-UNet. |
| `modular_le_admm5_pre` | Pre-UNet → 5 ADMM steps. |
| `modular_le_admm5_post` | 5 ADMM steps → post-UNet. |
| `modular_le_fista5_prepost` | Pre-UNet → 5 FISTA steps → post-UNet. |
**GAN super-resolution**
| Subfolder | Description |
|---|---|
| `admm100_bsrgan` | ADMM-100 + pretrained BSRGAN x4. |
| `admm100_bsrgan_finetune` | ADMM-100 + BSRGAN x4 fine-tuned on DigiCam. |
## Usage
```python
from transformers import AutoModel
model = AutoModel.from_pretrained(
"timofeiiz/lensless-computational-imaging",
subfolder="modular_le_admm5_prepost",
trust_remote_code=True,
)
recon = model(lensless, psf)
```
|