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..8478bb7938956f596131791457a0e2b66f99a716 --- /dev/null +++ b/.hydra/config.yaml @@ -0,0 +1,104 @@ +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: trainable_inv + 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 + trainable_inv: + K: 0.0001 + pre_process: + network: null + depth: 2 + nc: null + delay: null + freeze: null + unfreeze: null + post_process: + network: DruNet + depth: 2 + nc: null + delay: null + freeze: null + unfreeze: null + train_last_layer: false +trainable_mask: + mask_type: TrainablePSF + 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..4662a0fddd2c85469b850f495f9bc23d6703cd91 --- /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.method=trainable_inv + - trainable_mask.mask_type=TrainablePSF + - reconstruction.post_process.network=DruNet + - training.epoch=25 + job: + name: train_unrolled + chdir: true + override_dirname: reconstruction.method=trainable_inv,reconstruction.post_process.network=DruNet,trainable_mask.mask_type=TrainablePSF,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-25/17-43-29 + 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..fc17f56159d775ae31b1893df94ccd60a15af59f --- /dev/null +++ b/.hydra/overrides.yaml @@ -0,0 +1,4 @@ +- reconstruction.method=trainable_inv +- trainable_mask.mask_type=TrainablePSF +- reconstruction.post_process.network=DruNet +- training.epoch=25 diff --git a/eval_recon/.DS_Store b/eval_recon/.DS_Store new file mode 100644 index 0000000000000000000000000000000000000000..5ac4fbed9ce3496bacf017c9b8b94dcff12c47f7 Binary files /dev/null and b/eval_recon/.DS_Store differ diff --git a/eval_recon/0/0.png b/eval_recon/0/0.png new file mode 100644 index 0000000000000000000000000000000000000000..b65fc344c0df3f7e00f9d6c6c55c41872734dfb7 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..928b829825701a595d77350c0278df93fd53c93b 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..8b2e9349461c836288169cca99e0a3346879b187 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..f6fd45e407b6fd6d7b8d56c717934b0e93dc9d1e 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..c482d05fdfc33a148aca921f672fe1628b9d93f8 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..32f49ef7aac6f6a2dcbf69d3d21760d7cc5a8db1 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..271530c472d1b38f4a1c81172a381ef57e8bc5a3 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..3117300c9c9682d5bc758239400f2fc4d550387f 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..834efab5d210a9b1b288d868ac8e8598a9489739 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..5cd9e37007dac983a053a636afcf63274c7d3681 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..14c3721ff5c71da35d8309aa593e086e1bfd5680 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..188c0b8ad9b39030ff4f2a5c2f227b248e124907 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..ce579d49bcb0014a7ff856f852c6b7a39b428992 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..3fba54aac1b9ea72b5d89503bdb392b19ff2244c 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..21f280711f0c79bf2ecaf88bba95875085c524d7 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..343168161370a9c463da5c39e2758891a92f40fa 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..58437e8242907586aa0ff28f3547401c52ee7716 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..9fcf9b9b4fc2ee3111df07c6ce957d4fe0013495 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..4dc2f76adce1c1be41cce166d02be6ef34fe55a1 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..e74afecfb8885c08a2ec472ccb892cbb5951ec85 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..f603cb9a26d4b96800299467844a2b6d1c740d3c 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..ecf30a0d3d5a86d6c4af5ab05a1eaacc2c36fa76 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..471c5726673407a1b79c81a2acc6b157f0326f81 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..85918e886951ffb9297dd603a9086448f29c6e32 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..721165eb3c30f999c6f7e1321d9739104c9c4972 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..f6e378fc5c438746828fac04013b7d4fe29d82b1 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..fe6b70af761d7851923740f85d9ed91bcdd1e5d1 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..21e75ec27940b815d1ec8dff0e2269c1554cf604 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..94ddd1b8c75aa2d00e8c5ac01d73c02c3245a2c0 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..0b9bff242787e8c7961e0d6e69b4bc60d96d30e9 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..b940faba408bb21bdf49e6670420bf9e02b12644 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..c3f3c69992ab3c28ffcc4cbb6d99da0167f8ec38 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..215343053ee79106bef52e2235a269fd03873402 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..2dc42533ed86b2fb26c1ff4def6844cb29120aff 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..3dab27aba85fa2df511c33aa1913f2b0acd9e055 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..507164132e46d545589b8d6db2c16784e9179a06 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..b441d3672e2efadc0c01a0775beaf34b828049aa 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..450dc85acb1a90b139e6310a3d67ee811ee6ea11 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..ccd2768a11949599765eb27ccd361bad77acc4a7 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..0079e67d440d93cd2ef0b476a69f1cb45c22ae3d 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..da222e34075280e1cf4af78ea7defb2e7af0184b 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..a21ffe52f47ed7eff75eddfcfdca6092779d7fa0 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..a7d8357ae60c183c833f7bd55a321831ada9c648 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..4dbda494bc176a27ffd09429c6a11c351b160d37 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..8cc3047737af4d225650ae20b531f1e6c096e278 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..b68b97c4e5d79d6c7bc24e98191163df4d3a8c14 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..cd89b9d387cb86aa35ed4b1f4c3e37668aee4277 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..8dc819a98b20fa05a7343102d5e48b1ec5febcf9 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..37243573a472bdfde2f4716652dc353bbabf0285 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..7e03d31879c6cd9a0abb03ab3a2f2bc5e6f95786 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..471c0689bdd5730d2f8d78944ee3af50ec2dc199 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..764ae433664b0b37a33cc1b1e54cd8e322e60dfd 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..a48e031f73d285a5c3bb43b47da3cade945bf5f6 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..4162451a9945ea6974a7904e2d5425e7f7fd6b5d 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..62940ae1496d9f778bafca7a1e44bb68a06eb5f2 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..61232f991c55d9c0588f142dd025b22bcd68e3d4 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..df9486c0d9e98fdfb06fde223ef0a64f9b0d6203 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..685af73f0dfb0d7043c12e75de38358ade15d509 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..c6d28d7077533872cffcf19d4f1ff5b6a1ca0fd4 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..4cc9a1d6d461bfc1a46a0f3549ba2b5fa351b55b 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..16f656bf26c09b5169fbf2980f80becf66675a38 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..8394575ad1197b598747311b6d593c27653b460d 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..8fcb9e1f595867235a6ef25b65c48474c8d85a58 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..7d435f53a015f75a3ab3f8c9e8e4c989f0c2b034 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..b46bb38f621dc1874b5f5a75be2e4a9be94f2888 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..49d5cf9df6e590319a7cdc85855157e8397ad662 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..74ab4061b177b419fc540e9b8e2296c2ffa78ceb 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..d20a50e41b0607c1e540865152f5b90407a1a754 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..007e1b424410c9ee042396d5061dc18bd0d46638 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..38003172ded2c623265ea21b0f1d90465b775261 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..b22d1ecae5fec8009113aff2403d9a8a1485479e 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..a5162bbe583f9dd978b52fdff2d5559d806882e6 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..7abaa564c78ee92891bcba806b2c8c327808ba48 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..5754461e53de0ee74aca591af1ab042961da4209 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..563fe8b46ff3cbd961cb6fa0fb4499b5ea070f15 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..786a909a7a9b6d5a85f60b160a3e3d1b0c6aa607 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..c758e130d4f11fb2a8e890bb94d5efba61fc6e66 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..10e5e5240fb5f64f9923cc40c20790d1cd18973e 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..4a91ccf5e9504a908f2c3567152fa2b5171f36ea 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..e56f1481b0fbad244c624034c6c96e1987c702ed 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..9b85ee19e5ab3a383bbaf0b18702fda787da7ff5 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..8f39bb25080e2770c04f49ccd5d104140eb85b1c 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..c004401eb0d40524ad78f8525600e1ad56ef2034 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..6654d665ed2456143b5af94e9199e7d2a78a80ee 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..cf5df77b2833220d9f6c4630a506e5cfc13d7b7b 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..7eb4192a58a7f4d0e962202ffcb9a9f76b69d5e6 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..16b602ee293a7cbb98c02cffb06516a28821f420 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..458dcdff01f0578e6218b0eb4fdaa0cb28de5fb0 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..90449745dac3e6339f2e77ab84a7dc40cc3ab544 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..1c30facc933c8c2b948bda76610c3a9d5886928f 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..5c54a5f19d0b14add6ba437c8d68297cce61b024 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..c8fc7995f5f95ce4f4aa6505863da572034a1a43 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..3a947e1d54b3d1bd0c46cf0962ac069261f63176 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..ad0ac3fb648078221e74faec66a596a8f2a74c43 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..912281d08528201964f4edf5da3f6613cdbb3e64 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..efc9ad8ddf4a2bf026c40f4e63adc12db3be1651 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..cbfaa0785d5ca0e6ae7c1d8cd82606473903a3eb 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..1b5b7c88de15cd52ace3ff9ed86f7fa9741ce3c4 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..4aa4627897d27fa45217850ef794a92b93cec466 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..1c2eb7763ef111513b0a14a67c702a1f9b7281be 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..493b9c7f7c497f229086b677213a546886885fa0 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..fadb2b3e45ca5fd8bba9a129cb50be49c7c29ca0 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..0bb4ed306fe84f3e1a0116341c2f4a81ba051a00 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..09856e679aeb08f0e5adc456648f27981013935f 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..c0f379df0c60e9553f864ce99e5a105457c9f3bb 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..551c74c20dee00bfc39c9f3c36e228f297015ae9 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..81ca569d2a618053b47df590b795273959ef2fc0 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..eececca8c2e1dba4e25df5d0b456c9a4eb18e727 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..bd694e50d88f7175ca06b1d597e37a1e8a2f1842 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..5bdfa74368e8ac5e85eb5e18382d8ae08b8c32ae 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..d2917a68e658e372c8f6f91c7f9a7df9dd382308 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..997333396012073aa4f496efa394aa1d14cd4a88 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..a8b91d4b28d493960eb33e019672530671f7bbcd 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..e9d1570e1bf5e2108a50a1c4d429b22d03278447 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..6d30444052390c6a9bb71b225eb2ab7610f560db 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..2f740067a7190625fea1fe6839325fac04a52701 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..73ff15b8a700c8b51866755debde85cc7b828a89 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..c5c2fe9451f4e576e6d87f1213a7aff6c8d17ffc 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..b95cf1bef21d928b4e6c005114050fbe35c6e858 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..e4e67a7b26362bf31ab917ad5102449f221486ff 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..7b18668c88fdead91dd267294124e6a72ffbb231 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..8d533a695db78b815d39c41bfd032cbef5e20f53 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..9795e0ff820441897cd456798206947cc612cec3 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..3cc6f8185f2b12977719537086b75f2860901042 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..68538c99f16ca3608c767f8aadcd677a343c6f24 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..3b406ef2f053e4e19ed1430e6e584de5d7f8619b 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..6f5ab837fe0a3465d52c948490bc7ad4b00d1158 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..8283610451fcb2a88a21b4a984f4ba629fcaff1a 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..896653f40eee987a5a0386ac0f505e148a0be6da 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..29fffbd1cbe4c60ef06367b6bda92bd0111dc1fc Binary files /dev/null and b/eval_recon/9/4.png differ diff --git a/mask_epochBEST.npy b/mask_epochBEST.npy new file mode 100644 index 0000000000000000000000000000000000000000..577cd55620d2cafdc5afc00f5aa6882c8594aba5 --- /dev/null +++ b/mask_epochBEST.npy @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:983eee1b3ca8f0a66b33b5681bfbe498d0ac172769a2517eb29e03693a57faba +size 388928 diff --git a/mask_optim_epochBEST.pt b/mask_optim_epochBEST.pt new file mode 100644 index 0000000000000000000000000000000000000000..349947c0fafcd95d7d570c7f53b1aac4852b5465 --- /dev/null +++ b/mask_optim_epochBEST.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:b1773e7f122b2c364c9908bfecd77da2eb2781291776927109c9ab91e80f39c7 +size 779236 diff --git a/metrics.json b/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..d59e550a1d9684992db376a0023b18077abfb87c --- /dev/null +++ b/metrics.json @@ -0,0 +1,258 @@ +{ + "LOSS": [ + -1, + 0.42506172037124607, + 0.3715379193723203, + 0.3513330962657927, + 0.33664788734912926, + 0.32508560132483544, + 0.31761638724307173, + 0.3107886923700576, + 0.3049010163992641, + 0.29941335070629865, + 0.2952556682278715, + 0.2908715750028692, + 0.2871121711333593, + 0.2836354361226164, + 0.27971938257416235, + 0.2761586771557726, + 0.27307987784842636, + 0.269839598769943, + 0.2671173327267169, + 0.26434651042024365, + 0.2615345354080202, + 0.2592328493793806, + 0.25651818836232126, + 0.2541421487977099, + 0.251509060263634, + 0.24944406007726969 + ], + "LOSS_TEST": [ + 0.8817427600873634, + 0.3911167828924954, + 0.36132846454158424, + 0.34794712722767146, + 0.3297273882525042, + 0.33049568712711336, + 0.3197627825755626, + 0.3098387015331536, + 0.3074369425419718, + 0.3005863538896665, + 0.2974890483543277, + 0.2941137046040967, + 0.290157414926216, + 0.2909129729960114, + 0.2846108212461695, + 0.2850894840573892, + 0.28382050729822367, + 0.27896445008460435, + 0.27686190566513685, + 0.27418499733787033, + 0.27411451580934226, + 0.27319015124812723, + 0.27275842644274234, + 0.2712480311607942, + 0.2678498128731735 + ], + "MSE": [ + 0.10220292940735817, + 0.024723691530525685, + 0.01764351898804307, + 0.014283189442940056, + 0.012687594224698841, + 0.01504034087061882, + 0.012403487907722593, + 0.010030429298058152, + 0.009806539947167039, + 0.009120414867065847, + 0.00916629072278738, + 0.008947728392668068, + 0.008257289109751582, + 0.008758696736767888, + 0.008169838353060186, + 0.008402611813507974, + 0.007893917742185294, + 0.008037692443467677, + 0.007611328759230674, + 0.007293615522794426, + 0.008350841915234924, + 0.007456358801573515, + 0.00782252114266157, + 0.007512139878235758, + 0.006973667957354337, + 0.007694057673215866 + ], + "MAE": [ + 0.25300786033272743, + 0.08753189861774445, + 0.07182606883347034, + 0.06695936530828477, + 0.06161977462470532, + 0.069102883040905, + 0.06005643542855978, + 0.055133978240191936, + 0.054186781644821165, + 0.051850876845419405, + 0.05150843322277069, + 0.0511427466198802, + 0.04901222012937069, + 0.05031856644898653, + 0.048300758339464664, + 0.04936436712741852, + 0.048233044035732746, + 0.047971505597233774, + 0.04633164081722498, + 0.045605103895068166, + 0.04868423741310835, + 0.04610012032091618, + 0.04811137560755015, + 0.04641892924904823, + 0.0444523966871202, + 0.04809818860143423 + ], + "LPIPS_Vgg": [ + 0.7779993534088134, + 0.3600738856196404, + 0.3356810587644577, + 0.3243301922082901, + 0.3065659131109715, + 0.30440667539834976, + 0.29593536227941514, + 0.2878607977926731, + 0.28554745331406595, + 0.27908487990498543, + 0.27546654611825944, + 0.272397750467062, + 0.26898902773857114, + 0.2688723413646221, + 0.26317122131586074, + 0.26306485652923584, + 0.2621423216164112, + 0.25713752388954164, + 0.2552781166136265, + 0.2529674965143204, + 0.2516762213408947, + 0.25133099779486656, + 0.25052404329180716, + 0.24920787751674653, + 0.2463712391257286, + 0.2518304571509361 + ], + "LPIPS_Alex": [ + 0.7874577879905701, + 0.38550821006298064, + 0.33513267189264295, + 0.3068773880600929, + 0.2824141632020474, + 0.2725504489243031, + 0.26230364575982096, + 0.24702556684613228, + 0.24075234308838844, + 0.2342433078587055, + 0.2335679289698601, + 0.22968683674931525, + 0.2278301812708378, + 0.21962863832712173, + 0.22001352280378342, + 0.2205463695526123, + 0.21513023033738135, + 0.2132125949859619, + 0.2121272438764572, + 0.20117324694991112, + 0.20570312544703484, + 0.20081181690096855, + 0.2005457389354706, + 0.19821427419781684, + 0.19519629418849946, + 0.19907770559191704 + ], + "PSNR": [ + 9.927269506454468, + 16.123098306655884, + 17.59938841819763, + 18.52838647842407, + 19.041918201446535, + 18.294399070739747, + 19.13253080368042, + 20.07882122039795, + 20.17402603149414, + 20.490464153289796, + 20.46975305557251, + 20.569101409912108, + 20.926865253448486, + 20.666052894592283, + 20.97260311126709, + 20.841670207977295, + 21.1171453666687, + 21.040653228759766, + 21.28535743713379, + 21.472024211883546, + 20.872368354797363, + 21.375190105438232, + 21.14864995956421, + 21.339147090911865, + 21.666560153961182, + 21.24001905441284 + ], + "SSIM": [ + 0.21453354626893997, + 0.6007408636808396, + 0.654392392039299, + 0.6739414429664612, + 0.7043853408098221, + 0.7120751827955246, + 0.719514867067337, + 0.7372405737638473, + 0.7409637159109116, + 0.7432437640428543, + 0.751719828248024, + 0.7550811165571213, + 0.7605099564790726, + 0.7585941201448441, + 0.7662460827827453, + 0.7665924549102783, + 0.7680613201856613, + 0.7710643529891967, + 0.7774336498975754, + 0.7776895225048065, + 0.7778055906295777, + 0.7799130082130432, + 0.7818469029664993, + 0.7830485266447067, + 0.7862844932079315, + 0.7764676862955093 + ], + "ReconstructionError": [ + 146.57850433349608, + 128.24633537292482, + 114.94758346557617, + 102.84796516418457, + 90.13105144500733, + 83.79394836425782, + 79.79815441131592, + 73.13661018371582, + 71.00677650451661, + 67.35923751831055, + 61.94093242645263, + 62.197165946960446, + 60.304957733154296, + 56.30681941986084, + 56.41543670654297, + 52.64776905059814, + 50.61208019256592, + 50.480441398620606, + 49.128952960968014, + 49.65940837860107, + 48.145922050476074, + 45.36034027099609, + 45.61148950576782, + 45.97268684387207, + 45.432257118225095, + 43.916062088012694 + ], + "n_iter": 1, + "algorithm": "trainable_inv_DruNet", + "metric_for_best_model": null, + "best_epoch": 24, + "best_eval_score": 0.2678498128731735 +} \ No newline at end of file diff --git a/psf_epochBEST.npy b/psf_epochBEST.npy new file mode 100644 index 0000000000000000000000000000000000000000..87404ec1c50cad33930012818c805fc8a663e7ce --- /dev/null +++ b/psf_epochBEST.npy @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:cffd4cad6cbcb8156213d4af04d5e04543d05b45537aea2d16567ca170ce177d +size 388928 diff --git a/psf_epochBEST.png b/psf_epochBEST.png new file mode 100644 index 0000000000000000000000000000000000000000..b66b625b038ec0da96a8f9d9ea7a91f0f981f7f6 Binary files /dev/null and b/psf_epochBEST.png differ diff --git a/psf_epochBEST_plot.png b/psf_epochBEST_plot.png new file mode 100644 index 0000000000000000000000000000000000000000..33b9a88e77354ea27fd11de0354a1333c0089499 Binary files /dev/null and b/psf_epochBEST_plot.png differ diff --git a/psf_initial.png b/psf_initial.png new file mode 100644 index 0000000000000000000000000000000000000000..db9b977e219226c82f5304ae6d606b8320b3fefc Binary files /dev/null and b/psf_initial.png differ diff --git a/psf_initial_plot.png b/psf_initial_plot.png new file mode 100644 index 0000000000000000000000000000000000000000..a4d599e71eae08d7e88314bf7a249dbc9bbd0fa3 Binary files /dev/null and b/psf_initial_plot.png differ diff --git a/recon_epochBEST b/recon_epochBEST new file mode 100644 index 0000000000000000000000000000000000000000..8a23ca3019b6b217a83c8603bcc3a01dd43c7818 --- /dev/null +++ b/recon_epochBEST @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1aeb015bcf6f795cd7dff2ee5d8817f0da8f01f04561d265d9d1c79a5455606d +size 130983999 diff --git a/train_unrolled.log b/train_unrolled.log new file mode 100644 index 0000000000000000000000000000000000000000..f85ab564af9e89e611817dd84f7f1210ac1759ca --- /dev/null +++ b/train_unrolled.log @@ -0,0 +1,60 @@ +[2024-01-25 17:43:29,458][__main__][INFO] - Using GPU for training. +[2024-01-25 17:43:31,679][__main__][INFO] - Reconstruction a few images with ADMM... +[2024-01-25 17:43:33,262][__main__][INFO] - Train test size : 24000 +[2024-01-25 17:43:33,263][__main__][INFO] - Test test size : 999 +[2024-01-25 17:43:33,767][__main__][INFO] - Training model with 32738161 parameters +[2024-01-25 17:43:33,768][__main__][INFO] - Setup time : 0.5050997734069824 s +[2024-01-25 17:43:33,768][__main__][INFO] - PSF shape : torch.Size([1, 135, 240, 3]) +[2024-01-25 17:43:33,768][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-25/17-43-29 +[2024-01-25 17:44:56,823][__main__][INFO] - Epoch 0 with learning rate [0.0001] +[2024-01-25 18:24:48,679][__main__][INFO] - loss : 0.42506172037124607 +[2024-01-25 18:26:09,293][__main__][INFO] - Epoch 1 with learning rate [0.0001] +[2024-01-25 19:05:58,393][__main__][INFO] - loss : 0.3715379193723203 +[2024-01-25 19:07:18,251][__main__][INFO] - Epoch 2 with learning rate [0.0001] +[2024-01-25 19:46:56,253][__main__][INFO] - loss : 0.3513330962657927 +[2024-01-25 19:48:16,570][__main__][INFO] - Epoch 3 with learning rate [0.0001] +[2024-01-25 20:07:22,239][__main__][INFO] - loss : 0.33664788734912926 +[2024-01-25 20:07:38,938][__main__][INFO] - Epoch 4 with learning rate [0.0001] +[2024-01-25 20:20:37,098][__main__][INFO] - loss : 0.32508560132483544 +[2024-01-25 20:20:54,012][__main__][INFO] - Epoch 5 with learning rate [0.0001] +[2024-01-25 20:33:53,297][__main__][INFO] - loss : 0.31761638724307173 +[2024-01-25 20:34:10,909][__main__][INFO] - Epoch 6 with learning rate [0.0001] +[2024-01-25 20:47:08,539][__main__][INFO] - loss : 0.3107886923700576 +[2024-01-25 20:47:26,074][__main__][INFO] - Epoch 7 with learning rate [0.0001] +[2024-01-25 21:00:22,379][__main__][INFO] - loss : 0.3049010163992641 +[2024-01-25 21:00:39,054][__main__][INFO] - Epoch 8 with learning rate [0.0001] +[2024-01-25 21:13:35,620][__main__][INFO] - loss : 0.29941335070629865 +[2024-01-25 21:13:53,196][__main__][INFO] - Epoch 9 with learning rate [0.0001] +[2024-01-25 21:26:51,404][__main__][INFO] - loss : 0.2952556682278715 +[2024-01-25 21:27:08,669][__main__][INFO] - Epoch 10 with learning rate [0.0001] +[2024-01-25 21:40:05,099][__main__][INFO] - loss : 0.2908715750028692 +[2024-01-25 21:40:21,494][__main__][INFO] - Epoch 11 with learning rate [0.0001] +[2024-01-25 21:53:14,970][__main__][INFO] - loss : 0.2871121711333593 +[2024-01-25 21:53:31,394][__main__][INFO] - Epoch 12 with learning rate [0.0001] +[2024-01-25 22:06:25,855][__main__][INFO] - loss : 0.2836354361226164 +[2024-01-25 22:06:42,240][__main__][INFO] - Epoch 13 with learning rate [0.0001] +[2024-01-25 22:19:35,705][__main__][INFO] - loss : 0.27971938257416235 +[2024-01-25 22:19:52,161][__main__][INFO] - Epoch 14 with learning rate [0.0001] +[2024-01-25 22:32:43,573][__main__][INFO] - loss : 0.2761586771557726 +[2024-01-25 22:32:59,523][__main__][INFO] - Epoch 15 with learning rate [0.0001] +[2024-01-25 22:45:53,160][__main__][INFO] - loss : 0.27307987784842636 +[2024-01-25 22:46:11,070][__main__][INFO] - Epoch 16 with learning rate [0.0001] +[2024-01-25 22:59:05,409][__main__][INFO] - loss : 0.269839598769943 +[2024-01-25 22:59:23,087][__main__][INFO] - Epoch 17 with learning rate [0.0001] +[2024-01-25 23:12:17,260][__main__][INFO] - loss : 0.2671173327267169 +[2024-01-25 23:12:34,912][__main__][INFO] - Epoch 18 with learning rate [0.0001] +[2024-01-25 23:25:26,143][__main__][INFO] - loss : 0.26434651042024365 +[2024-01-25 23:25:43,128][__main__][INFO] - Epoch 19 with learning rate [0.0001] +[2024-01-25 23:38:36,653][__main__][INFO] - loss : 0.2615345354080202 +[2024-01-25 23:38:54,382][__main__][INFO] - Epoch 20 with learning rate [0.0001] +[2024-01-25 23:51:46,856][__main__][INFO] - loss : 0.2592328493793806 +[2024-01-25 23:52:03,763][__main__][INFO] - Epoch 21 with learning rate [0.0001] +[2024-01-26 00:04:57,997][__main__][INFO] - loss : 0.25651818836232126 +[2024-01-26 00:05:14,157][__main__][INFO] - Epoch 22 with learning rate [0.0001] +[2024-01-26 00:18:08,774][__main__][INFO] - loss : 0.2541421487977099 +[2024-01-26 00:18:25,562][__main__][INFO] - Epoch 23 with learning rate [0.0001] +[2024-01-26 00:31:21,350][__main__][INFO] - loss : 0.251509060263634 +[2024-01-26 00:31:37,978][__main__][INFO] - Epoch 24 with learning rate [0.0001] +[2024-01-26 00:44:32,539][__main__][INFO] - loss : 0.24944406007726969 +[2024-01-26 00:44:48,670][__main__][INFO] - Train time [hour] : 7.020398301680883 h +[2024-01-26 00:44:48,670][__main__][INFO] - Results saved in /home/bezzam/LenslessPiCam/outputs/2024-01-25/17-43-29