|
seed: 0 |
|
files: |
|
dataset: /scratch/bezzam/celeba/celeba_adafruit_random_30cm_2mm_20231004_26K |
|
celeba_root: /scratch/bezzam |
|
psf: rpi_hq_adafruit_psf_2mm/raw_data_rgb.png |
|
diffusercam_psf: true |
|
n_files: null |
|
downsample: 2 |
|
test_size: 0.15 |
|
vertical_shift: -117 |
|
horizontal_shift: -25 |
|
crop: |
|
vertical: |
|
- 0 |
|
- 525 |
|
horizontal: |
|
- 265 |
|
- 695 |
|
torch: true |
|
torch_device: cuda |
|
test_idx: |
|
- 0 |
|
- 1 |
|
- 2 |
|
- 3 |
|
- 4 |
|
eval_disp_idx: |
|
- 0 |
|
- 1 |
|
- 2 |
|
- 3 |
|
- 4 |
|
display: |
|
plot: true |
|
gamma: null |
|
save: true |
|
reconstruction: |
|
method: unrolled_admm |
|
unrolled_fista: |
|
n_iter: 20 |
|
tk: 1 |
|
learn_tk: true |
|
unrolled_admm: |
|
n_iter: 10 |
|
mu1: 0.0001 |
|
mu2: 0.0001 |
|
mu3: 0.0001 |
|
tau: 0.0002 |
|
pre_process: |
|
network: null |
|
depth: 2 |
|
nc: null |
|
post_process: |
|
network: null |
|
depth: 2 |
|
nc: |
|
- 32 |
|
- 64 |
|
- 128 |
|
- 256 |
|
trainable_mask: |
|
mask_type: AdafruitLCD |
|
initial_value: adafruit_random_pattern_20231004_174047.npy |
|
grayscale: false |
|
mask_lr: 0.001 |
|
L1_strength: false |
|
train_mask_vals: true |
|
train_color_filter: true |
|
ap_center: |
|
- 58 |
|
- 76 |
|
ap_shape: |
|
- 19 |
|
- 25 |
|
rotate: 0 |
|
vertical_shift: -80 |
|
horizontal_shift: -60 |
|
slm: adafruit |
|
sensor: rpi_hq |
|
flipud: true |
|
waveprop: false |
|
scene2mask: 0.3 |
|
mask2sensor: 0.002 |
|
target: object_plane |
|
simulation: |
|
grayscale: false |
|
output_dim: null |
|
object_height: 0.33 |
|
flip: true |
|
random_shift: false |
|
random_vflip: false |
|
random_hflip: false |
|
random_rotate: false |
|
scene2mask: 0.25 |
|
mask2sensor: 0.002 |
|
sensor: rpi_hq |
|
snr_db: null |
|
downsample: null |
|
quantize: false |
|
max_val: 255 |
|
training: |
|
batch_size: 4 |
|
epoch: 25 |
|
eval_batch_size: 15 |
|
metric_for_best_model: null |
|
save_every: 5 |
|
skip_NAN: true |
|
crop_preloss: true |
|
optimizer: |
|
type: Adam |
|
lr: 0.0001 |
|
slow_start: false |
|
step: false |
|
gamma: 0.1 |
|
loss: l2 |
|
lpips: 1.0 |
|
|