File size: 1,856 Bytes
78ed824 |
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 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 |
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
|