|
seed: 0 |
|
start_delay: null |
|
files: |
|
dataset: /scratch/bezzam/DiffuserCam_mirflickr/dataset |
|
celeba_root: null |
|
psf: data/psf/diffusercam_psf.tiff |
|
diffusercam_psf: true |
|
n_files: null |
|
downsample: 2 |
|
test_size: 0.15 |
|
input_snr: 10 |
|
vertical_shift: null |
|
horizontal_shift: null |
|
crop: null |
|
torch: true |
|
torch_device: cuda |
|
measure: null |
|
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 |
|
skip_unrolled: false |
|
unrolled_fista: |
|
n_iter: 20 |
|
tk: 1 |
|
learn_tk: true |
|
unrolled_admm: |
|
n_iter: 20 |
|
mu1: 0.0001 |
|
mu2: 0.0001 |
|
mu3: 0.0001 |
|
tau: 0.0002 |
|
pre_process: |
|
network: null |
|
depth: 2 |
|
nc: null |
|
delay: null |
|
freeze: null |
|
unfreeze: null |
|
post_process: |
|
network: null |
|
depth: 2 |
|
nc: null |
|
delay: null |
|
freeze: null |
|
unfreeze: null |
|
train_last_layer: false |
|
trainable_mask: |
|
mask_type: null |
|
initial_value: psf |
|
grayscale: false |
|
mask_lr: 0.001 |
|
L1_strength: 1.0 |
|
target: object_plane |
|
simulation: |
|
grayscale: false |
|
output_dim: null |
|
object_height: 0.04 |
|
flip: true |
|
random_shift: false |
|
random_vflip: 0.5 |
|
random_hflip: 0.5 |
|
random_rotate: false |
|
scene2mask: 0.1 |
|
mask2sensor: 0.009 |
|
sensor: rpi_hq |
|
snr_db: 10 |
|
downsample: 8 |
|
quantize: false |
|
max_val: 255 |
|
training: |
|
batch_size: 8 |
|
epoch: 25 |
|
eval_batch_size: 10 |
|
metric_for_best_model: null |
|
save_every: null |
|
skip_NAN: true |
|
clip_grad: 1.0 |
|
crop_preloss: false |
|
optimizer: |
|
type: Adam |
|
lr: 0.0001 |
|
slow_start: false |
|
step: false |
|
gamma: 0.1 |
|
loss: l2 |
|
lpips: 1.0 |
|
unrolled_output_factor: false |
|
|