Upload 156 files

.hydra/config.yaml

@@ -1,37 +1,33 @@
 seed: 0
 start_delay: null
 files:
-  dataset: /scratch/bezzam/celeba/celeba_adafruit_random_30cm_2mm_20231004_26K
-  celeba_root: /scratch/bezzam
-  psf: outputs/2023-10-25/09-40-03/adafruit_random_pattern_20231004_174047_SIM_psf.png
+  dataset: bezzam/DigiCam-CelebA-26K
+  celeba_root: null
+  psf: data/psf/diffusercam_psf.tiff
   diffusercam_psf: true
+  huggingface_dataset: true
+  huggingface_psf: psf_simulated.png
+  split_seed: 0
   n_files: null
   downsample: 2
   test_size: 0.15
-  vertical_shift: -117
-  horizontal_shift: -25
-  crop:
-    vertical:
-    - 0
-    - 525
-    horizontal:
-    - 265
-    - 695
+  input_snr: null
+  vertical_shift: null
+  horizontal_shift: null
+  crop: null
+  image_res: null
+  extra_eval: null
+  rotate: true
+  save_psf: false
 torch: true
-torch_device: cuda
+torch_device: cuda:0
 measure: null
-test_idx:
-- 0
-- 1
-- 2
-- 3
-- 4
 eval_disp_idx:
 - 0
-- 1
 - 2
 - 3
 - 4
+- 9
 display:
   plot: true
   gamma: null
@@ -39,6 +35,9 @@ save: true
 reconstruction:
   method: unrolled_admm
   skip_unrolled: false
+  init_processors: null
+  init_pre: true
+  init_post: true
   unrolled_fista:
     n_iter: 20
     tk: 1
@@ -49,6 +48,8 @@ reconstruction:
     mu2: 0.0001
     mu3: 0.0001
     tau: 0.0002
+  trainable_inv:
+    K: 0.0001
   pre_process:
     network: UnetRes
     depth: 4
@@ -77,28 +78,29 @@ trainable_mask:
   initial_value: psf
   grayscale: false
   mask_lr: 0.001
+  optimizer: Adam
   L1_strength: 1.0
 target: object_plane
 simulation:
   grayscale: false
   output_dim: null
-  object_height: 0.33
+  object_height: 0.04
   flip: true
   random_shift: false
-  random_vflip: false
-  random_hflip: false
+  random_vflip: 0.5
+  random_hflip: 0.5
   random_rotate: false
-  scene2mask: 0.25
-  mask2sensor: 0.002
+  scene2mask: 0.1
+  mask2sensor: 0.009
   sensor: rpi_hq
-  snr_db: null
-  downsample: null
+  snr_db: 10
+  downsample: 8
   quantize: false
   max_val: 255
 training:
   batch_size: 4
   epoch: 25
-  eval_batch_size: 16
+  eval_batch_size: 4
   metric_for_best_model: null
   save_every: null
   skip_NAN: true
@@ -112,3 +114,29 @@ optimizer:
   gamma: 0.1
 loss: l2
 lpips: 1.0
+unrolled_output_factor: false
+device_ids:
+- 0
+- 1
+- 2
+- 3
+alignment:
+  crop:
+    vertical:
+    - 0
+    - 525
+    horizontal:
+    - 265
+    - 695
+  simulation:
+    scene2mask: 0.25
+    mask2sensor: 0.002
+    object_height: 0.33
+    sensor: rpi_hq
+    snr_db: null
+    downsample: null
+    random_vflip: false
+    random_hflip: false
+    quantize: false
+    vertical_shift: -117
+    horizontal_shift: -25

.hydra/hydra.yaml

@@ -113,21 +113,17 @@ hydra:
     - hydra.mode=RUN
     task:
     - files.downsample=2
-    - files.psf=outputs/2023-10-25/09-40-03/adafruit_random_pattern_20231004_174047_SIM_psf.png
     - training.batch_size=4
-    - reconstruction.pre_process.network=UnetRes
-    - reconstruction.pre_process.nc=[32,64,116,128]
-    - reconstruction.pre_process.depth=4
-    - reconstruction.post_process.network=UnetRes
-    - reconstruction.post_process.nc=[32,64,116,128]
-    - reconstruction.post_process.depth=4
+    - training.eval_batch_size=4
+    - torch_device=cuda:0
+    - device_ids=[0,1,2,3]
   job:
     name: train_unrolled
     chdir: true
-    override_dirname: files.downsample=2,files.psf=outputs/2023-10-25/09-40-03/adafruit_random_pattern_20231004_174047_SIM_psf.png,reconstruction.post_process.depth=4,reconstruction.post_process.nc=[32,64,116,128],reconstruction.post_process.network=UnetRes,reconstruction.pre_process.depth=4,reconstruction.pre_process.nc=[32,64,116,128],reconstruction.pre_process.network=UnetRes,training.batch_size=4
+    override_dirname: device_ids=[0,1,2,3],files.downsample=2,torch_device=cuda:0,training.batch_size=4,training.eval_batch_size=4
     id: ???
     num: ???
-    config_name: train_celeba_digicam
+    config_name: train_digicam_celeba
     env_set: {}
     env_copy: []
     config:
@@ -138,18 +134,18 @@ hydra:
   runtime:
     version: 1.3.2
     version_base: '1.3'
-    cwd: /home/bezzam/LenslessPiCam
+    cwd: /root/LenslessPiCam
     config_sources:
     - path: hydra.conf
       schema: pkg
       provider: hydra
-    - path: /home/bezzam/LenslessPiCam/configs
+    - path: /root/LenslessPiCam/configs
       schema: file
       provider: main
     - path: ''
       schema: structured
       provider: schema
-    output_dir: /home/bezzam/LenslessPiCam/outputs/2023-11-20/13-26-04
+    output_dir: /root/LenslessPiCam/outputs/2024-03-24/15-35-50
     choices:
       hydra/env: default
       hydra/callbacks: null

.hydra/overrides.yaml

@@ -1,9 +1,5 @@
 - files.downsample=2
-- files.psf=outputs/2023-10-25/09-40-03/adafruit_random_pattern_20231004_174047_SIM_psf.png
 - training.batch_size=4
-- reconstruction.pre_process.network=UnetRes
-- reconstruction.pre_process.nc=[32,64,116,128]
-- reconstruction.pre_process.depth=4
-- reconstruction.post_process.network=UnetRes
-- reconstruction.post_process.nc=[32,64,116,128]
-- reconstruction.post_process.depth=4
+- training.eval_batch_size=4
+- torch_device=cuda:0
+- device_ids=[0,1,2,3]