diff --git a/.gitattributes b/.gitattributes index a6344aac8c09253b3b630fb776ae94478aa0275b..e3f72c1d42ac3c7ac444291888758afcb00ebf8a 100644 --- a/.gitattributes +++ b/.gitattributes @@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text *.zip filter=lfs diff=lfs merge=lfs -text *.zst filter=lfs diff=lfs merge=lfs -text *tfevents* filter=lfs diff=lfs merge=lfs -text +recon_epochBEST filter=lfs diff=lfs merge=lfs -text diff --git a/.hydra/config.yaml b/.hydra/config.yaml new file mode 100644 index 0000000000000000000000000000000000000000..b7e2f698fee8064285c8d89653c8da19789437a3 --- /dev/null +++ b/.hydra/config.yaml @@ -0,0 +1,105 @@ +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: null + 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 + init_processors: Unet+U20+Unet_res0.01 + 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 + trainable_inv: + K: 0.0001 + pre_process: + network: UnetRes + depth: 2 + nc: null + delay: null + freeze: null + unfreeze: null + post_process: + network: UnetRes + 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 diff --git a/.hydra/hydra.yaml b/.hydra/hydra.yaml new file mode 100644 index 0000000000000000000000000000000000000000..a79b5b0f3a7b8dafb6b60f5fa27ff167fc14eedd --- /dev/null +++ b/.hydra/hydra.yaml @@ -0,0 +1,158 @@ +hydra: + run: + dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S} + sweep: + dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S} + subdir: ${hydra.job.num} + launcher: + _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher + sweeper: + _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper + max_batch_size: null + params: null + help: + app_name: ${hydra.job.name} + header: '${hydra.help.app_name} is powered by Hydra. + + ' + footer: 'Powered by Hydra (https://hydra.cc) + + Use --hydra-help to view Hydra specific help + + ' + template: '${hydra.help.header} + + == Configuration groups == + + Compose your configuration from those groups (group=option) + + + $APP_CONFIG_GROUPS + + + == Config == + + Override anything in the config (foo.bar=value) + + + $CONFIG + + + ${hydra.help.footer} + + ' + hydra_help: + template: 'Hydra (${hydra.runtime.version}) + + See https://hydra.cc for more info. + + + == Flags == + + $FLAGS_HELP + + + == Configuration groups == + + Compose your configuration from those groups (For example, append hydra/job_logging=disabled + to command line) + + + $HYDRA_CONFIG_GROUPS + + + Use ''--cfg hydra'' to Show the Hydra config. + + ' + hydra_help: ??? + hydra_logging: + version: 1 + formatters: + simple: + format: '[%(asctime)s][HYDRA] %(message)s' + handlers: + console: + class: logging.StreamHandler + formatter: simple + stream: ext://sys.stdout + root: + level: INFO + handlers: + - console + loggers: + logging_example: + level: DEBUG + disable_existing_loggers: false + job_logging: + version: 1 + formatters: + simple: + format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s' + handlers: + console: + class: logging.StreamHandler + formatter: simple + stream: ext://sys.stdout + file: + class: logging.FileHandler + formatter: simple + filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log + root: + level: INFO + handlers: + - console + - file + disable_existing_loggers: false + env: {} + mode: RUN + searchpath: [] + callbacks: {} + output_subdir: .hydra + overrides: + hydra: + - hydra.mode=RUN + task: + - reconstruction.pre_process.network=UnetRes + - reconstruction.post_process.network=UnetRes + - reconstruction.init_processors=Unet+U20+Unet_res0.01 + - training.epoch=25 + job: + name: train_unrolled + chdir: true + override_dirname: reconstruction.init_processors=Unet+U20+Unet_res0.01,reconstruction.post_process.network=UnetRes,reconstruction.pre_process.network=UnetRes,training.epoch=25 + id: ??? + num: ??? + config_name: train_unrolledADMM + env_set: {} + env_copy: [] + config: + override_dirname: + kv_sep: '=' + item_sep: ',' + exclude_keys: [] + runtime: + version: 1.3.2 + version_base: '1.3' + cwd: /home/bezzam/LenslessPiCam + config_sources: + - path: hydra.conf + schema: pkg + provider: hydra + - path: /home/bezzam/LenslessPiCam/configs + schema: file + provider: main + - path: '' + schema: structured + provider: schema + output_dir: /home/bezzam/LenslessPiCam/outputs/2024-01-27/16-04-47 + choices: + hydra/env: default + hydra/callbacks: null + hydra/job_logging: default + hydra/hydra_logging: default + hydra/hydra_help: default + hydra/help: default + hydra/sweeper: basic + hydra/launcher: basic + hydra/output: default + verbose: false diff --git a/.hydra/overrides.yaml b/.hydra/overrides.yaml new file mode 100644 index 0000000000000000000000000000000000000000..7b16769aaddb1922806bdeef20f3a02257195989 --- /dev/null +++ b/.hydra/overrides.yaml @@ -0,0 +1,4 @@ +- reconstruction.pre_process.network=UnetRes +- reconstruction.post_process.network=UnetRes +- reconstruction.init_processors=Unet+U20+Unet_res0.01 +- training.epoch=25 diff --git a/eval_recon/0/0.png b/eval_recon/0/0.png new file mode 100644 index 0000000000000000000000000000000000000000..2d6039522f717dadee8389001b31089acde4532f Binary files /dev/null and b/eval_recon/0/0.png differ diff --git a/eval_recon/0/1.png b/eval_recon/0/1.png new file mode 100644 index 0000000000000000000000000000000000000000..76a55ad0b7f22a5ee428588b1f8998ad7be08edc Binary files /dev/null and b/eval_recon/0/1.png differ diff --git a/eval_recon/0/2.png b/eval_recon/0/2.png new file mode 100644 index 0000000000000000000000000000000000000000..0855166436437a32505ab9ffaa8d7ffa7f99b15c Binary files /dev/null and b/eval_recon/0/2.png differ diff --git a/eval_recon/0/3.png b/eval_recon/0/3.png new file mode 100644 index 0000000000000000000000000000000000000000..7aa726d071c7bfe187042b363f81248d53670938 Binary files /dev/null and b/eval_recon/0/3.png differ diff --git a/eval_recon/0/4.png b/eval_recon/0/4.png new file mode 100644 index 0000000000000000000000000000000000000000..3e082b809e5df7d67611316182eb545989721403 Binary files /dev/null and b/eval_recon/0/4.png differ diff --git a/eval_recon/1/0.png b/eval_recon/1/0.png new file mode 100644 index 0000000000000000000000000000000000000000..68627a95e7c3af52fa9698c48255e482cb4b2479 Binary files /dev/null and b/eval_recon/1/0.png differ diff --git a/eval_recon/1/1.png b/eval_recon/1/1.png new file mode 100644 index 0000000000000000000000000000000000000000..616e043e1fc59d6bb55a41876158f9195e1c249b Binary files /dev/null and b/eval_recon/1/1.png differ diff --git a/eval_recon/1/2.png b/eval_recon/1/2.png new file mode 100644 index 0000000000000000000000000000000000000000..7dcca180fa188274c7c8c074a700e9ca02773e47 Binary files /dev/null and b/eval_recon/1/2.png differ diff --git a/eval_recon/1/3.png b/eval_recon/1/3.png new file mode 100644 index 0000000000000000000000000000000000000000..e60a9095dc24a535b30b56fa6a1d59704e073ebc Binary files /dev/null and b/eval_recon/1/3.png differ diff --git a/eval_recon/1/4.png b/eval_recon/1/4.png new file mode 100644 index 0000000000000000000000000000000000000000..1bf2d123a16bb705469bb0c5c0ba888dec9b1753 Binary files /dev/null and b/eval_recon/1/4.png differ diff --git a/eval_recon/10/0.png b/eval_recon/10/0.png new file mode 100644 index 0000000000000000000000000000000000000000..d4021715329c1734eefa90be46374f0d0b6c600c Binary files /dev/null and b/eval_recon/10/0.png differ diff --git a/eval_recon/10/1.png b/eval_recon/10/1.png new file mode 100644 index 0000000000000000000000000000000000000000..b2c337931dbaf04663a43ea3ea831e3b023f3408 Binary files /dev/null and b/eval_recon/10/1.png differ diff --git a/eval_recon/10/2.png b/eval_recon/10/2.png new file mode 100644 index 0000000000000000000000000000000000000000..a9e9076afb86a06ba42f31b22cb2800c0f261643 Binary files /dev/null and b/eval_recon/10/2.png differ diff --git a/eval_recon/10/3.png b/eval_recon/10/3.png new file mode 100644 index 0000000000000000000000000000000000000000..02b8196011927dd1a736706f3dd44c13de374c4f Binary files /dev/null and b/eval_recon/10/3.png differ diff --git a/eval_recon/10/4.png b/eval_recon/10/4.png new file mode 100644 index 0000000000000000000000000000000000000000..b4e605e5acc77d3055a5f828cce970db52b6ea71 Binary files /dev/null and b/eval_recon/10/4.png differ diff --git a/eval_recon/11/0.png b/eval_recon/11/0.png new file mode 100644 index 0000000000000000000000000000000000000000..5d1b7544bc6b393859d72edfa9296610d8821217 Binary files /dev/null and b/eval_recon/11/0.png differ diff --git a/eval_recon/11/1.png b/eval_recon/11/1.png new file mode 100644 index 0000000000000000000000000000000000000000..a01ecab6d614c03dc439fe734cbe40dc8deb99fd Binary files /dev/null and b/eval_recon/11/1.png differ diff --git a/eval_recon/11/2.png b/eval_recon/11/2.png new file mode 100644 index 0000000000000000000000000000000000000000..e2b429248d2c964bfe9ce1936a1497251739fc2a Binary files /dev/null and b/eval_recon/11/2.png differ diff --git a/eval_recon/11/3.png b/eval_recon/11/3.png new file mode 100644 index 0000000000000000000000000000000000000000..3dbe52ce2aec03578aa5869bdcd02faceb26cc03 Binary files /dev/null and b/eval_recon/11/3.png differ diff --git a/eval_recon/11/4.png b/eval_recon/11/4.png new file mode 100644 index 0000000000000000000000000000000000000000..6ad2205d20a4e9b7afec39153c0492b346cb544b Binary files /dev/null and b/eval_recon/11/4.png differ diff --git a/eval_recon/12/0.png b/eval_recon/12/0.png new file mode 100644 index 0000000000000000000000000000000000000000..2ecdf8b32d99f2ed49df180a8b6fb85e5c90e4cd Binary files /dev/null and b/eval_recon/12/0.png differ diff --git a/eval_recon/12/1.png b/eval_recon/12/1.png new file mode 100644 index 0000000000000000000000000000000000000000..8f131f50ed8aec6d2328ada30ee0df0591f2a531 Binary files /dev/null and b/eval_recon/12/1.png differ diff --git a/eval_recon/12/2.png b/eval_recon/12/2.png new file mode 100644 index 0000000000000000000000000000000000000000..2cae3381f9176eda7fedee219f4fb2069ada14d2 Binary files /dev/null and b/eval_recon/12/2.png differ diff --git a/eval_recon/12/3.png b/eval_recon/12/3.png new file mode 100644 index 0000000000000000000000000000000000000000..e18a23bf279fcfa7b5c64887fd521d0a0d8b4b74 Binary files /dev/null and b/eval_recon/12/3.png differ diff --git a/eval_recon/12/4.png b/eval_recon/12/4.png new file mode 100644 index 0000000000000000000000000000000000000000..699ab0adbe48600d6db2083ed4f0af63b9e3a5e3 Binary files /dev/null and b/eval_recon/12/4.png differ diff --git a/eval_recon/13/0.png b/eval_recon/13/0.png new file mode 100644 index 0000000000000000000000000000000000000000..fecebc25ea2e7363f0add3bb810fa0f325e306ff Binary files /dev/null and b/eval_recon/13/0.png differ diff --git a/eval_recon/13/1.png b/eval_recon/13/1.png new file mode 100644 index 0000000000000000000000000000000000000000..496f73f38f9f9c2875d255357259944fc59a1232 Binary files /dev/null and b/eval_recon/13/1.png differ diff --git a/eval_recon/13/2.png b/eval_recon/13/2.png new file mode 100644 index 0000000000000000000000000000000000000000..9d11b2a65333e53ba710e38842b3732bb93a4a08 Binary files /dev/null and b/eval_recon/13/2.png differ diff --git a/eval_recon/13/3.png b/eval_recon/13/3.png new file mode 100644 index 0000000000000000000000000000000000000000..5528f9eb92268cc914a6d787575f5a7fb31130f3 Binary files /dev/null and b/eval_recon/13/3.png differ diff --git a/eval_recon/13/4.png b/eval_recon/13/4.png new file mode 100644 index 0000000000000000000000000000000000000000..40dd965255686b160de6a03eafb69142484633d2 Binary files /dev/null and b/eval_recon/13/4.png differ diff --git a/eval_recon/14/0.png b/eval_recon/14/0.png new file mode 100644 index 0000000000000000000000000000000000000000..76779db97f4e6ba667ba741abe474ced8cbdde63 Binary files /dev/null and b/eval_recon/14/0.png differ diff --git a/eval_recon/14/1.png b/eval_recon/14/1.png new file mode 100644 index 0000000000000000000000000000000000000000..6dad13874af2602a7c02999bf55802cae144746b Binary files /dev/null and b/eval_recon/14/1.png differ diff --git a/eval_recon/14/2.png b/eval_recon/14/2.png new file mode 100644 index 0000000000000000000000000000000000000000..05caef9249101b1a66ad7e85446907b9dae7dc7d Binary files /dev/null and b/eval_recon/14/2.png differ diff --git a/eval_recon/14/3.png b/eval_recon/14/3.png new file mode 100644 index 0000000000000000000000000000000000000000..fb59639c13aed80603511b2be7629f4a4bffb78c Binary files /dev/null and b/eval_recon/14/3.png differ diff --git a/eval_recon/14/4.png b/eval_recon/14/4.png new file mode 100644 index 0000000000000000000000000000000000000000..21dc0b36af037b2cd262eca33ad4611d077fd1d3 Binary files /dev/null and b/eval_recon/14/4.png differ diff --git a/eval_recon/15/0.png b/eval_recon/15/0.png new file mode 100644 index 0000000000000000000000000000000000000000..e0aeb610b84fd39d984a80073a26c9e7955b6d49 Binary files /dev/null and b/eval_recon/15/0.png differ diff --git a/eval_recon/15/1.png b/eval_recon/15/1.png new file mode 100644 index 0000000000000000000000000000000000000000..1146cd15264868945067611d0aeb6e44e967f001 Binary files /dev/null and b/eval_recon/15/1.png differ diff --git a/eval_recon/15/2.png b/eval_recon/15/2.png new file mode 100644 index 0000000000000000000000000000000000000000..68e3733114613077cfbcc47feb1ac09c2b239a67 Binary files /dev/null and b/eval_recon/15/2.png differ diff --git a/eval_recon/15/3.png b/eval_recon/15/3.png new file mode 100644 index 0000000000000000000000000000000000000000..a2911d6d811730fd5fa6f18003b093c9a4a35fd6 Binary files /dev/null and b/eval_recon/15/3.png differ diff --git a/eval_recon/15/4.png b/eval_recon/15/4.png new file mode 100644 index 0000000000000000000000000000000000000000..722a68a3b8debe762dd4df7471d75f0aaaee8165 Binary files /dev/null and b/eval_recon/15/4.png differ diff --git a/eval_recon/16/0.png b/eval_recon/16/0.png new file mode 100644 index 0000000000000000000000000000000000000000..353f913efac9eeab96b3517b69b1ac1fe293809d Binary files /dev/null and b/eval_recon/16/0.png differ diff --git a/eval_recon/16/1.png b/eval_recon/16/1.png new file mode 100644 index 0000000000000000000000000000000000000000..b75f3a8fac3984f3de6a3f415ea40fd5d1a93b2e Binary files /dev/null and b/eval_recon/16/1.png differ diff --git a/eval_recon/16/2.png b/eval_recon/16/2.png new file mode 100644 index 0000000000000000000000000000000000000000..f11415a195a62508cb23b996b1a49e52d91f0505 Binary files /dev/null and b/eval_recon/16/2.png differ diff --git a/eval_recon/16/3.png b/eval_recon/16/3.png new file mode 100644 index 0000000000000000000000000000000000000000..89380ebdb43ad9b3ef80ff940d497e4d2e5cc1b8 Binary files /dev/null and b/eval_recon/16/3.png differ diff --git a/eval_recon/16/4.png b/eval_recon/16/4.png new file mode 100644 index 0000000000000000000000000000000000000000..5e06705b3c11e86810c93520fecf864b8bc35d71 Binary files /dev/null and b/eval_recon/16/4.png differ diff --git a/eval_recon/17/0.png b/eval_recon/17/0.png new file mode 100644 index 0000000000000000000000000000000000000000..671f06299f0256ecfac20cbea30b85bd22232128 Binary files /dev/null and b/eval_recon/17/0.png differ diff --git a/eval_recon/17/1.png b/eval_recon/17/1.png new file mode 100644 index 0000000000000000000000000000000000000000..9d7d85051c440c9be3ade96c7fb57627b1f13cd5 Binary files /dev/null and b/eval_recon/17/1.png differ diff --git a/eval_recon/17/2.png b/eval_recon/17/2.png new file mode 100644 index 0000000000000000000000000000000000000000..ff47e374c940484aac507858734f1d49d7293029 Binary files /dev/null and b/eval_recon/17/2.png differ diff --git a/eval_recon/17/3.png b/eval_recon/17/3.png new file mode 100644 index 0000000000000000000000000000000000000000..30ba6620cb3f8a389850a715b91a5bac913ba6ce Binary files /dev/null and b/eval_recon/17/3.png differ diff --git a/eval_recon/17/4.png b/eval_recon/17/4.png new file mode 100644 index 0000000000000000000000000000000000000000..f24da978b04aa13dffce263fc2c0a57b6f231126 Binary files /dev/null and b/eval_recon/17/4.png differ diff --git a/eval_recon/18/0.png b/eval_recon/18/0.png new file mode 100644 index 0000000000000000000000000000000000000000..7b041179da458aeed2e8d3e866630c101d9fddf1 Binary files /dev/null and b/eval_recon/18/0.png differ diff --git a/eval_recon/18/1.png b/eval_recon/18/1.png new file mode 100644 index 0000000000000000000000000000000000000000..9f40ff202bd5f86bdc7662b5273202c4f188740b Binary files /dev/null and b/eval_recon/18/1.png differ diff --git a/eval_recon/18/2.png b/eval_recon/18/2.png new file mode 100644 index 0000000000000000000000000000000000000000..7c84d8b3b5ef93d82f6afeae5964c36e6b1e5546 Binary files /dev/null and b/eval_recon/18/2.png differ diff --git a/eval_recon/18/3.png b/eval_recon/18/3.png new file mode 100644 index 0000000000000000000000000000000000000000..050980efc0f249a03ee1924c1f5de0a6a44021d6 Binary files /dev/null and b/eval_recon/18/3.png differ diff --git a/eval_recon/18/4.png b/eval_recon/18/4.png new file mode 100644 index 0000000000000000000000000000000000000000..6418002c12874c4fbfb9e978896dccec5981d61a Binary files /dev/null and b/eval_recon/18/4.png differ diff --git a/eval_recon/19/0.png b/eval_recon/19/0.png new file mode 100644 index 0000000000000000000000000000000000000000..bd62ec882f180e9a3fbf01a5ddbcc7eda3d9781b Binary files /dev/null and b/eval_recon/19/0.png differ diff --git a/eval_recon/19/1.png b/eval_recon/19/1.png new file mode 100644 index 0000000000000000000000000000000000000000..3d24473a519e9100367b09dc8f1c9c9cbf04d3a7 Binary files /dev/null and b/eval_recon/19/1.png differ diff --git a/eval_recon/19/2.png b/eval_recon/19/2.png new file mode 100644 index 0000000000000000000000000000000000000000..123f6fb626eef0bf774b9e066443a7974bba92bd Binary files /dev/null and b/eval_recon/19/2.png differ diff --git a/eval_recon/19/3.png b/eval_recon/19/3.png new file mode 100644 index 0000000000000000000000000000000000000000..5dad55a54d9d428ab010f0bbf99a19a86c5f391a Binary files /dev/null and b/eval_recon/19/3.png differ diff --git a/eval_recon/19/4.png b/eval_recon/19/4.png new file mode 100644 index 0000000000000000000000000000000000000000..f738b7e72fdd7fa3c3cb62556b028a741d057e69 Binary files /dev/null and b/eval_recon/19/4.png differ diff --git a/eval_recon/2/0.png b/eval_recon/2/0.png new file mode 100644 index 0000000000000000000000000000000000000000..da7e55e85a5bf8d5ef544ddbe973b09df17a32fe Binary files /dev/null and b/eval_recon/2/0.png differ diff --git a/eval_recon/2/1.png b/eval_recon/2/1.png new file mode 100644 index 0000000000000000000000000000000000000000..eb1583ccc1a6d41cb2acd6deb3bfb04866cfc0e9 Binary files /dev/null and b/eval_recon/2/1.png differ diff --git a/eval_recon/2/2.png b/eval_recon/2/2.png new file mode 100644 index 0000000000000000000000000000000000000000..be604e1b3ec7a189cff498c65907d11d786c1db8 Binary files /dev/null and b/eval_recon/2/2.png differ diff --git a/eval_recon/2/3.png b/eval_recon/2/3.png new file mode 100644 index 0000000000000000000000000000000000000000..181d566087a8f724f8942dbcfc771b350f112a95 Binary files /dev/null and b/eval_recon/2/3.png differ diff --git a/eval_recon/2/4.png b/eval_recon/2/4.png new file mode 100644 index 0000000000000000000000000000000000000000..c6f36de878befc9b0f023c56fb1165bd56758e5f Binary files /dev/null and b/eval_recon/2/4.png differ diff --git a/eval_recon/20/0.png b/eval_recon/20/0.png new file mode 100644 index 0000000000000000000000000000000000000000..d596df4ac6b53da78932c681198dfb6c2bc40f3f Binary files /dev/null and b/eval_recon/20/0.png differ diff --git a/eval_recon/20/1.png b/eval_recon/20/1.png new file mode 100644 index 0000000000000000000000000000000000000000..62e232e5bbd84ca39926cc5685b44128ea22afae Binary files /dev/null and b/eval_recon/20/1.png differ diff --git a/eval_recon/20/2.png b/eval_recon/20/2.png new file mode 100644 index 0000000000000000000000000000000000000000..906286dd08f7d017bdcc741946b32eb5f7b8cacf Binary files /dev/null and b/eval_recon/20/2.png differ diff --git a/eval_recon/20/3.png b/eval_recon/20/3.png new file mode 100644 index 0000000000000000000000000000000000000000..c6562af8ee0511ed3e51936428262c2ec513da3a Binary files /dev/null and b/eval_recon/20/3.png differ diff --git a/eval_recon/20/4.png b/eval_recon/20/4.png new file mode 100644 index 0000000000000000000000000000000000000000..91dda84532392506116417919676df986623c3c4 Binary files /dev/null and b/eval_recon/20/4.png differ diff --git a/eval_recon/21/0.png b/eval_recon/21/0.png new file mode 100644 index 0000000000000000000000000000000000000000..20d6c40d2cf97330ce8220acd5d4c3feb58bf6dd Binary files /dev/null and b/eval_recon/21/0.png differ diff --git a/eval_recon/21/1.png b/eval_recon/21/1.png new file mode 100644 index 0000000000000000000000000000000000000000..ead64287e4d42707d67c95e82a81bd510552b6eb Binary files /dev/null and b/eval_recon/21/1.png differ diff --git a/eval_recon/21/2.png b/eval_recon/21/2.png new file mode 100644 index 0000000000000000000000000000000000000000..96e60572e42cb966341171d9d41916147b2417a4 Binary files /dev/null and b/eval_recon/21/2.png differ diff --git a/eval_recon/21/3.png b/eval_recon/21/3.png new file mode 100644 index 0000000000000000000000000000000000000000..de120baf3cf2561b307bbfa8903d714b51211183 Binary files /dev/null and b/eval_recon/21/3.png differ diff --git a/eval_recon/21/4.png b/eval_recon/21/4.png new file mode 100644 index 0000000000000000000000000000000000000000..42fc94bb040e1c09f0b1a5804043294f1bd5cda7 Binary files /dev/null and b/eval_recon/21/4.png differ diff --git a/eval_recon/22/0.png b/eval_recon/22/0.png new file mode 100644 index 0000000000000000000000000000000000000000..7db3d57ff62b380270ae8d1e5a0a299179e07284 Binary files /dev/null and b/eval_recon/22/0.png differ diff --git a/eval_recon/22/1.png b/eval_recon/22/1.png new file mode 100644 index 0000000000000000000000000000000000000000..fc1dd1c81f6dfc59fc981b30d8e822407adc630d Binary files /dev/null and b/eval_recon/22/1.png differ diff --git a/eval_recon/22/2.png b/eval_recon/22/2.png new file mode 100644 index 0000000000000000000000000000000000000000..a82382125ceae722d3da3e53a90f6a6c5b6496d9 Binary files /dev/null and b/eval_recon/22/2.png differ diff --git a/eval_recon/22/3.png b/eval_recon/22/3.png new file mode 100644 index 0000000000000000000000000000000000000000..be940c7bfdef6785f69db1cc9660398800cba97b Binary files /dev/null and b/eval_recon/22/3.png differ diff --git a/eval_recon/22/4.png b/eval_recon/22/4.png new file mode 100644 index 0000000000000000000000000000000000000000..48f19f1208f7879c2dbf1150c888e5175a488c3c Binary files /dev/null and b/eval_recon/22/4.png differ diff --git a/eval_recon/23/0.png b/eval_recon/23/0.png new file mode 100644 index 0000000000000000000000000000000000000000..fcbce85f9fd61be6c92a319c9b54d02289be17d6 Binary files /dev/null and b/eval_recon/23/0.png differ diff --git a/eval_recon/23/1.png b/eval_recon/23/1.png new file mode 100644 index 0000000000000000000000000000000000000000..a05285be1519fa634f436823bf22ff767872eae6 Binary files /dev/null and b/eval_recon/23/1.png differ diff --git a/eval_recon/23/2.png b/eval_recon/23/2.png new file mode 100644 index 0000000000000000000000000000000000000000..441bf4985c823bb80e3b82616c661f8368c5a19e Binary files /dev/null and b/eval_recon/23/2.png differ diff --git a/eval_recon/23/3.png b/eval_recon/23/3.png new file mode 100644 index 0000000000000000000000000000000000000000..bb071f37bc4270985e6e5f583e93a4a6c61c4356 Binary files /dev/null and b/eval_recon/23/3.png differ diff --git a/eval_recon/23/4.png b/eval_recon/23/4.png new file mode 100644 index 0000000000000000000000000000000000000000..e2b9801c46d301737495943e2858dc04889a81c5 Binary files /dev/null and b/eval_recon/23/4.png differ diff --git a/eval_recon/24/0.png b/eval_recon/24/0.png new file mode 100644 index 0000000000000000000000000000000000000000..56f75d6fafd7191c3798e7473d1818c122761d2e Binary files /dev/null and b/eval_recon/24/0.png differ diff --git a/eval_recon/24/1.png b/eval_recon/24/1.png new file mode 100644 index 0000000000000000000000000000000000000000..60e2ebbcd37318c0b79bc75e47df4ff46aad5553 Binary files /dev/null and b/eval_recon/24/1.png differ diff --git a/eval_recon/24/2.png b/eval_recon/24/2.png new file mode 100644 index 0000000000000000000000000000000000000000..935c92a7fef499c5d5fa79a94d477c5890ed7460 Binary files /dev/null and b/eval_recon/24/2.png differ diff --git a/eval_recon/24/3.png b/eval_recon/24/3.png new file mode 100644 index 0000000000000000000000000000000000000000..bae52883ccc6e216859da72df86e5c141bad9035 Binary files /dev/null and b/eval_recon/24/3.png differ diff --git a/eval_recon/24/4.png b/eval_recon/24/4.png new file mode 100644 index 0000000000000000000000000000000000000000..e1d6635722a5c4145f68c0dd3aaea1294fa90d93 Binary files /dev/null and b/eval_recon/24/4.png differ diff --git a/eval_recon/25/0.png b/eval_recon/25/0.png new file mode 100644 index 0000000000000000000000000000000000000000..b036e1191057cfbf73ac408dc437f42b7ad5812a Binary files /dev/null and b/eval_recon/25/0.png differ diff --git a/eval_recon/25/1.png b/eval_recon/25/1.png new file mode 100644 index 0000000000000000000000000000000000000000..03c3a7b2dbd451e75c10a350751a7dd511902114 Binary files /dev/null and b/eval_recon/25/1.png differ diff --git a/eval_recon/25/2.png b/eval_recon/25/2.png new file mode 100644 index 0000000000000000000000000000000000000000..8b4be4a54171684f6012b924e9d717dbc7d404b4 Binary files /dev/null and b/eval_recon/25/2.png differ diff --git a/eval_recon/25/3.png b/eval_recon/25/3.png new file mode 100644 index 0000000000000000000000000000000000000000..882d3fb36d1be40d9bc185fd10c6db3b338826ca Binary files /dev/null and b/eval_recon/25/3.png differ diff --git a/eval_recon/25/4.png b/eval_recon/25/4.png new file mode 100644 index 0000000000000000000000000000000000000000..6e98f8a0c5fb53f6a47602597ee5c41563cc0527 Binary files /dev/null and b/eval_recon/25/4.png differ diff --git a/eval_recon/3/0.png b/eval_recon/3/0.png new file mode 100644 index 0000000000000000000000000000000000000000..f50ba08b16bdf4497afbfa18ab2bb0313c9cf39c Binary files /dev/null and b/eval_recon/3/0.png differ diff --git a/eval_recon/3/1.png b/eval_recon/3/1.png new file mode 100644 index 0000000000000000000000000000000000000000..3f5d9fa9e29c67f0fc376bc368006c17ca511e40 Binary files /dev/null and b/eval_recon/3/1.png differ diff --git a/eval_recon/3/2.png b/eval_recon/3/2.png new file mode 100644 index 0000000000000000000000000000000000000000..a4768585d5b60b76d333eee7cb54a49caf43f76e Binary files /dev/null and b/eval_recon/3/2.png differ diff --git a/eval_recon/3/3.png b/eval_recon/3/3.png new file mode 100644 index 0000000000000000000000000000000000000000..3343129e7fecd54318b3a7fc525f76c1c17f6c80 Binary files /dev/null and b/eval_recon/3/3.png differ diff --git a/eval_recon/3/4.png b/eval_recon/3/4.png new file mode 100644 index 0000000000000000000000000000000000000000..e5915edda95420c045637f10052a41441bb19d77 Binary files /dev/null and b/eval_recon/3/4.png differ diff --git a/eval_recon/4/0.png b/eval_recon/4/0.png new file mode 100644 index 0000000000000000000000000000000000000000..8a7cd1eb0fbb51b2d0c1890dd67b45ea7dea1255 Binary files /dev/null and b/eval_recon/4/0.png differ diff --git a/eval_recon/4/1.png b/eval_recon/4/1.png new file mode 100644 index 0000000000000000000000000000000000000000..1f1bbfe4aa0e3297960b7623aeeb4d845d0319e3 Binary files /dev/null and b/eval_recon/4/1.png differ diff --git a/eval_recon/4/2.png b/eval_recon/4/2.png new file mode 100644 index 0000000000000000000000000000000000000000..6ae28145cc9aa280e5da0265f94ed60475328f54 Binary files /dev/null and b/eval_recon/4/2.png differ diff --git a/eval_recon/4/3.png b/eval_recon/4/3.png new file mode 100644 index 0000000000000000000000000000000000000000..b77aa68c4dd6db570d78fd03737632d0964418af Binary files /dev/null and b/eval_recon/4/3.png differ diff --git a/eval_recon/4/4.png b/eval_recon/4/4.png new file mode 100644 index 0000000000000000000000000000000000000000..8c32ee62ce190e1d6c59679401330cf5d2118df7 Binary files /dev/null and b/eval_recon/4/4.png differ diff --git a/eval_recon/5/0.png b/eval_recon/5/0.png new file mode 100644 index 0000000000000000000000000000000000000000..b71df25392fbf7b26d634bd490d2070f5e40c4d4 Binary files /dev/null and b/eval_recon/5/0.png differ diff --git a/eval_recon/5/1.png b/eval_recon/5/1.png new file mode 100644 index 0000000000000000000000000000000000000000..b859b16df53a406d45da17e96ca6ca63476f1348 Binary files /dev/null and b/eval_recon/5/1.png differ diff --git a/eval_recon/5/2.png b/eval_recon/5/2.png new file mode 100644 index 0000000000000000000000000000000000000000..8aaf7d73b7215777ec0abd6456426e887db2a88f Binary files /dev/null and b/eval_recon/5/2.png differ diff --git a/eval_recon/5/3.png b/eval_recon/5/3.png new file mode 100644 index 0000000000000000000000000000000000000000..b638d3415daa09cca4a6decc91843e2d27a42caf Binary files /dev/null and b/eval_recon/5/3.png differ diff --git a/eval_recon/5/4.png b/eval_recon/5/4.png new file mode 100644 index 0000000000000000000000000000000000000000..e5a6ff5276db85ba78b20e419fd8da9381f1ae15 Binary files /dev/null and b/eval_recon/5/4.png differ diff --git a/eval_recon/6/0.png b/eval_recon/6/0.png new file mode 100644 index 0000000000000000000000000000000000000000..6e1c4a07c5aa3c8f307ff05cfdb5c3ed31e51d0f Binary files /dev/null and b/eval_recon/6/0.png differ diff --git a/eval_recon/6/1.png b/eval_recon/6/1.png new file mode 100644 index 0000000000000000000000000000000000000000..24744adebbfcbf24790623ce427035dd376cd202 Binary files /dev/null and b/eval_recon/6/1.png differ diff --git a/eval_recon/6/2.png b/eval_recon/6/2.png new file mode 100644 index 0000000000000000000000000000000000000000..5cb2318b534ac1a602f58ab3c4bb119ed03eb353 Binary files /dev/null and b/eval_recon/6/2.png differ diff --git a/eval_recon/6/3.png b/eval_recon/6/3.png new file mode 100644 index 0000000000000000000000000000000000000000..8deaf9972befc4da109a26591790f545505621e1 Binary files /dev/null and b/eval_recon/6/3.png differ diff --git a/eval_recon/6/4.png b/eval_recon/6/4.png new file mode 100644 index 0000000000000000000000000000000000000000..b4db68778a7c4221902dd3931de7470cbcf935ef Binary files /dev/null and b/eval_recon/6/4.png differ diff --git a/eval_recon/7/0.png b/eval_recon/7/0.png new file mode 100644 index 0000000000000000000000000000000000000000..e4266dbf1ab31d0a5bc3fa938954edd18c9c9b95 Binary files /dev/null and b/eval_recon/7/0.png differ diff --git a/eval_recon/7/1.png b/eval_recon/7/1.png new file mode 100644 index 0000000000000000000000000000000000000000..1df524fab9095872d58cb921b9d9bd24ae0685f1 Binary files /dev/null and b/eval_recon/7/1.png differ diff --git a/eval_recon/7/2.png b/eval_recon/7/2.png new file mode 100644 index 0000000000000000000000000000000000000000..b4c52a939c540d718104d69f924505217e8fa71f Binary files /dev/null and b/eval_recon/7/2.png differ diff --git a/eval_recon/7/3.png b/eval_recon/7/3.png new file mode 100644 index 0000000000000000000000000000000000000000..f16f1f8e9471d99947658550f3e653bf28aa649b Binary files /dev/null and b/eval_recon/7/3.png differ diff --git a/eval_recon/7/4.png b/eval_recon/7/4.png new file mode 100644 index 0000000000000000000000000000000000000000..273d71e93938ecf25dabda4bb5fcbd9c07409f4f Binary files /dev/null and b/eval_recon/7/4.png differ diff --git a/eval_recon/8/0.png b/eval_recon/8/0.png new file mode 100644 index 0000000000000000000000000000000000000000..29a2a73e771e7cce71fea3e2302b2effaf26caab Binary files /dev/null and b/eval_recon/8/0.png differ diff --git a/eval_recon/8/1.png b/eval_recon/8/1.png new file mode 100644 index 0000000000000000000000000000000000000000..bca3c0268905fbfa276af45ba04a6fa66d323253 Binary files /dev/null and b/eval_recon/8/1.png differ diff --git a/eval_recon/8/2.png b/eval_recon/8/2.png new file mode 100644 index 0000000000000000000000000000000000000000..88e20cd44d3326de5418dfb7b0efea49d4c96866 Binary files /dev/null and b/eval_recon/8/2.png differ diff --git a/eval_recon/8/3.png b/eval_recon/8/3.png new file mode 100644 index 0000000000000000000000000000000000000000..7db5628d6f1d1c91f21ea4bf3c01b09f33d83dc7 Binary files /dev/null and b/eval_recon/8/3.png differ diff --git a/eval_recon/8/4.png b/eval_recon/8/4.png new file mode 100644 index 0000000000000000000000000000000000000000..a44f3bf095a056b2392330e4cc28125ee77e8435 Binary files /dev/null and b/eval_recon/8/4.png differ diff --git a/eval_recon/9/0.png b/eval_recon/9/0.png new file mode 100644 index 0000000000000000000000000000000000000000..93c8b30fd4a16efc7c66dae441d97a9b690bfa6a Binary files /dev/null and b/eval_recon/9/0.png differ diff --git a/eval_recon/9/1.png b/eval_recon/9/1.png new file mode 100644 index 0000000000000000000000000000000000000000..66682e6696bfce9bb0f4d76da388038288922be3 Binary files /dev/null and b/eval_recon/9/1.png differ diff --git a/eval_recon/9/2.png b/eval_recon/9/2.png new file mode 100644 index 0000000000000000000000000000000000000000..d02683121148858b4efb9646312d0dee41647470 Binary files /dev/null and b/eval_recon/9/2.png differ diff --git a/eval_recon/9/3.png b/eval_recon/9/3.png new file mode 100644 index 0000000000000000000000000000000000000000..74f715caf3e40ccafd6ec1776ee6dab5a15ddc44 Binary files /dev/null and b/eval_recon/9/3.png differ diff --git a/eval_recon/9/4.png b/eval_recon/9/4.png new file mode 100644 index 0000000000000000000000000000000000000000..71b601bd86ce8a5f30b3dded623684b4bb3ef58e Binary files /dev/null and b/eval_recon/9/4.png differ diff --git a/metrics.json b/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..4b58e0e6ef72b474efbd6faf226a531c6c713836 --- /dev/null +++ b/metrics.json @@ -0,0 +1,258 @@ +{ + "LOSS": [ + -1, + 0.2904620627711222, + 0.22702715347210572, + 0.20910989487667875, + 0.19793190091351665, + 0.19041558116177718, + 0.1843203121076023, + 0.17984511188914334, + 0.1747767227689429, + 0.17105758504569524, + 0.16789209530999252, + 0.16390000306318184, + 0.1614447848250468, + 0.15819583714753419, + 0.15576349579791215, + 0.15290123852839085, + 0.15053067220995822, + 0.1479654023870823, + 0.14615798941502997, + 0.14384513116627923, + 0.14138763200740037, + 0.13947315612186997, + 0.13772385247051708, + 0.13618720066547416, + 0.13420913152893368, + 0.13257588320473815 + ], + "LOSS_TEST": [ + 0.7122386718541384, + 0.25616559890098867, + 0.22045374520821498, + 0.20226813803659754, + 0.20004395346855744, + 0.2002770006400533, + 0.19773065970744938, + 0.1921139817289077, + 0.19093214967404493, + 0.18864378368365578, + 0.18638234318466856, + 0.18510060949600302, + 0.18314215918304397, + 0.18294088363531047, + 0.18008692332776263, + 0.1824416391388513, + 0.18201389895402825, + 0.17974508447456175, + 0.18188104424276388, + 0.17888290699571371, + 0.18118437442579308, + 0.17858581123757175, + 0.1807673674693797, + 0.1833950234029908, + 0.17768361592665313 + ], + "MSE": [ + 0.13666503198444843, + 0.010344344945624471, + 0.005145480267237872, + 0.0039504816685803235, + 0.003990786036010832, + 0.0042831001919694245, + 0.00409238354768604, + 0.0038885788968764244, + 0.0035916993871796875, + 0.0033320452820044012, + 0.003324389171320945, + 0.003153113111620769, + 0.0032306185853667557, + 0.0032935960579197855, + 0.0031174372998066247, + 0.0031854085461236535, + 0.0032642533036414535, + 0.003131820986745879, + 0.0030491415003780276, + 0.0030843637511134146, + 0.003101654831552878, + 0.0032285626034718007, + 0.003000588553259149, + 0.003067803439917043, + 0.003036263082176447, + 0.0029985348193440585 + ], + "MAE": [ + 0.22102899491786956, + 0.0606456957384944, + 0.04204888427630067, + 0.03599428121000528, + 0.03668503822758794, + 0.03817405194044113, + 0.03684940364211798, + 0.03605561738833785, + 0.035094968546181916, + 0.03315059231594205, + 0.03337324924767018, + 0.03216561228036881, + 0.03272947691380978, + 0.03295297915115952, + 0.03182576166465879, + 0.03218148430809378, + 0.03276462519541383, + 0.03197158988565207, + 0.03160699211061001, + 0.03174515316262841, + 0.0323286204226315, + 0.0326522215642035, + 0.03134238531813025, + 0.031791045255959036, + 0.031292502637952566, + 0.031325569432228806 + ], + "LPIPS_Vgg": [ + 0.5755736398696899, + 0.24582125395536422, + 0.2153082649409771, + 0.1983176563680172, + 0.1960531674325466, + 0.19599390044808387, + 0.19363827615976334, + 0.18822540283203126, + 0.18734045028686525, + 0.18531173840165138, + 0.18305795401334762, + 0.18194749638438223, + 0.17991154059767722, + 0.17964728757739068, + 0.176969486027956, + 0.17925623059272766, + 0.1787496456503868, + 0.17661326348781586, + 0.17883190274238586, + 0.17579854324460029, + 0.17808271959424019, + 0.17535724863409996, + 0.17776677891612053, + 0.18032721996307374, + 0.1746473528444767, + 0.17695050552487374 + ], + "LPIPS_Alex": [ + 0.543559196293354, + 0.189056978225708, + 0.14252271719276904, + 0.12410450823605061, + 0.1171067188680172, + 0.11575740061700344, + 0.12202975034713745, + 0.11546769559383392, + 0.10303237251937389, + 0.10123716861009598, + 0.10064447425305843, + 0.1021320304274559, + 0.10062085129320622, + 0.09898947082459926, + 0.10142605938017368, + 0.09860097624361515, + 0.10371465742588043, + 0.09734301239252091, + 0.09909691967070103, + 0.10004736825823785, + 0.09654236003756524, + 0.09877089068293571, + 0.09457112178206444, + 0.09687645725905895, + 0.09595264606177807, + 0.0930198709666729 + ], + "PSNR": [ + 8.74937551021576, + 19.939402408599854, + 22.985370502471923, + 24.13875463485718, + 24.081052646636962, + 23.776674900054932, + 23.983199005126952, + 24.189998207092284, + 24.552509059906004, + 24.87554485321045, + 24.889332046508787, + 25.118235664367674, + 25.004834995269775, + 24.927384777069094, + 25.166181640625, + 25.065324268341065, + 24.957329540252687, + 25.144738235473632, + 25.273034000396727, + 25.202464923858642, + 25.187352199554443, + 25.00268989562988, + 25.33769952774048, + 25.238393020629882, + 25.28297519683838, + 25.340473079681395 + ], + "SSIM": [ + 0.2533037443459034, + 0.7073441606760025, + 0.7967582511901855, + 0.8243284672498703, + 0.8254066574573516, + 0.8259247654676437, + 0.8307303994894027, + 0.8379514753818512, + 0.8376998621225357, + 0.8417959421873092, + 0.8429395037889481, + 0.8463619381189347, + 0.8464679116010666, + 0.8463981705904007, + 0.8491350489854813, + 0.8465408766269684, + 0.847718660235405, + 0.8483422720432281, + 0.8486094522476196, + 0.8519145011901855, + 0.8494201874732972, + 0.8511925911903382, + 0.8485481029748917, + 0.8487890857458115, + 0.852072394490242, + 0.8516828465461731 + ], + "ReconstructionError": [ + 30.843561706542967, + 39.53229068756104, + 41.109977645874025, + 39.47583535194397, + 39.0723627948761, + 38.73791766166687, + 40.86325629234314, + 43.09621934890747, + 40.913657131195066, + 40.66791760444641, + 41.66260498046875, + 41.14400005340576, + 41.844298725128176, + 43.084584369659424, + 40.65788519859314, + 42.210078735351566, + 43.678738822937014, + 42.429266633987424, + 42.276074657440184, + 43.44919179916382, + 42.42494161605835, + 46.53905281066895, + 46.804638385772705, + 45.45742338180542, + 46.89881883621216, + 46.41274198532105 + ], + "n_iter": 20, + "algorithm": "UnetRes_d2_unrolled_admm_UnetRes_d2", + "metric_for_best_model": null, + "best_epoch": 24, + "best_eval_score": 0.17768361592665313 +} \ No newline at end of file diff --git a/recon_epochBEST b/recon_epochBEST new file mode 100644 index 0000000000000000000000000000000000000000..578d891cea18171bbfbf2080e7dc0a36e7c2ec80 --- /dev/null +++ b/recon_epochBEST @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:d5256f2d135bebd21da2de789f9a558bf1d13cb2b024ac164bf516f079679ea6 +size 136121211 diff --git a/train_unrolled.log b/train_unrolled.log new file mode 100644 index 0000000000000000000000000000000000000000..c2c478006f0fb76ced2baeecc667f5da45724cef --- /dev/null +++ b/train_unrolled.log @@ -0,0 +1,60 @@ +[2024-01-27 16:04:47,153][__main__][INFO] - Using GPU for training. +[2024-01-27 16:04:49,168][__main__][INFO] - Reconstruction a few images with ADMM... +[2024-01-27 16:04:50,427][__main__][INFO] - Train test size : 24000 +[2024-01-27 16:04:50,427][__main__][INFO] - Test test size : 999 +[2024-01-27 16:04:50,980][__main__][INFO] - Training model with 34021330 parameters +[2024-01-27 16:04:50,980][__main__][INFO] - Setup time : 0.5530400276184082 s +[2024-01-27 16:04:50,980][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) +[2024-01-27 16:04:50,980][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-27/16-04-47 +[2024-01-27 16:05:21,669][__main__][INFO] - Epoch 0 with learning rate [0.0001] +[2024-01-27 16:31:56,246][__main__][INFO] - loss : 0.2904620627711222 +[2024-01-27 16:32:24,617][__main__][INFO] - Epoch 1 with learning rate [0.0001] +[2024-01-27 16:58:56,452][__main__][INFO] - loss : 0.22702715347210572 +[2024-01-27 16:59:25,164][__main__][INFO] - Epoch 2 with learning rate [0.0001] +[2024-01-27 17:25:54,003][__main__][INFO] - loss : 0.20910989487667875 +[2024-01-27 17:26:21,353][__main__][INFO] - Epoch 3 with learning rate [0.0001] +[2024-01-27 17:52:50,291][__main__][INFO] - loss : 0.19793190091351665 +[2024-01-27 17:53:18,505][__main__][INFO] - Epoch 4 with learning rate [0.0001] +[2024-01-27 18:19:46,649][__main__][INFO] - loss : 0.19041558116177718 +[2024-01-27 18:20:14,007][__main__][INFO] - Epoch 5 with learning rate [0.0001] +[2024-01-27 18:46:44,381][__main__][INFO] - loss : 0.1843203121076023 +[2024-01-27 18:47:11,732][__main__][INFO] - Epoch 6 with learning rate [0.0001] +[2024-01-27 19:13:41,149][__main__][INFO] - loss : 0.17984511188914334 +[2024-01-27 19:14:08,607][__main__][INFO] - Epoch 7 with learning rate [0.0001] +[2024-01-27 19:40:49,725][__main__][INFO] - loss : 0.1747767227689429 +[2024-01-27 19:41:17,730][__main__][INFO] - Epoch 8 with learning rate [0.0001] +[2024-01-27 20:07:45,613][__main__][INFO] - loss : 0.17105758504569524 +[2024-01-27 20:08:13,562][__main__][INFO] - Epoch 9 with learning rate [0.0001] +[2024-01-27 20:34:42,224][__main__][INFO] - loss : 0.16789209530999252 +[2024-01-27 20:35:11,232][__main__][INFO] - Epoch 10 with learning rate [0.0001] +[2024-01-27 21:01:38,963][__main__][INFO] - loss : 0.16390000306318184 +[2024-01-27 21:02:07,254][__main__][INFO] - Epoch 11 with learning rate [0.0001] +[2024-01-27 21:28:35,031][__main__][INFO] - loss : 0.1614447848250468 +[2024-01-27 21:29:03,631][__main__][INFO] - Epoch 12 with learning rate [0.0001] +[2024-01-27 21:55:31,473][__main__][INFO] - loss : 0.15819583714753419 +[2024-01-27 21:55:58,636][__main__][INFO] - Epoch 13 with learning rate [0.0001] +[2024-01-27 22:22:27,409][__main__][INFO] - loss : 0.15576349579791215 +[2024-01-27 22:22:54,695][__main__][INFO] - Epoch 14 with learning rate [0.0001] +[2024-01-27 22:49:22,344][__main__][INFO] - loss : 0.15290123852839085 +[2024-01-27 22:49:50,259][__main__][INFO] - Epoch 15 with learning rate [0.0001] +[2024-01-27 23:16:16,848][__main__][INFO] - loss : 0.15053067220995822 +[2024-01-27 23:16:44,718][__main__][INFO] - Epoch 16 with learning rate [0.0001] +[2024-01-27 23:43:14,301][__main__][INFO] - loss : 0.1479654023870823 +[2024-01-27 23:43:42,591][__main__][INFO] - Epoch 17 with learning rate [0.0001] +[2024-01-28 00:10:10,323][__main__][INFO] - loss : 0.14615798941502997 +[2024-01-28 00:10:37,835][__main__][INFO] - Epoch 18 with learning rate [0.0001] +[2024-01-28 00:37:34,062][__main__][INFO] - loss : 0.14384513116627923 +[2024-01-28 00:38:03,615][__main__][INFO] - Epoch 19 with learning rate [0.0001] +[2024-01-28 01:20:33,815][__main__][INFO] - loss : 0.14138763200740037 +[2024-01-28 01:21:50,613][__main__][INFO] - Epoch 20 with learning rate [0.0001] +[2024-01-28 02:09:39,269][__main__][INFO] - loss : 0.13947315612186997 +[2024-01-28 02:11:00,016][__main__][INFO] - Epoch 21 with learning rate [0.0001] +[2024-01-28 02:59:01,657][__main__][INFO] - loss : 0.13772385247051708 +[2024-01-28 03:00:20,895][__main__][INFO] - Epoch 22 with learning rate [0.0001] +[2024-01-28 03:48:16,466][__main__][INFO] - loss : 0.13618720066547416 +[2024-01-28 03:49:35,789][__main__][INFO] - Epoch 23 with learning rate [0.0001] +[2024-01-28 04:37:29,614][__main__][INFO] - loss : 0.13420913152893368 +[2024-01-28 04:38:53,070][__main__][INFO] - Epoch 24 with learning rate [0.0001] +[2024-01-28 05:26:58,011][__main__][INFO] - loss : 0.13257588320473815 +[2024-01-28 05:28:20,849][__main__][INFO] - Train time [hour] : 13.391367570559185 h +[2024-01-28 05:28:20,850][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-27/16-04-47