bezzam commited on
Commit
019e5b9
1 Parent(s): 7a791f2

Upload 156 files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ recon_epochBEST filter=lfs diff=lfs merge=lfs -text
.hydra/config.yaml ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ seed: 0
2
+ start_delay: null
3
+ files:
4
+ dataset: /scratch/bezzam/celeba/celeba_adafruit_random_30cm_2mm_20231004_26K
5
+ celeba_root: /scratch/bezzam
6
+ psf: rpi_hq_adafruit_psf_2mm/raw_data_rgb.png
7
+ diffusercam_psf: true
8
+ n_files: null
9
+ downsample: 2
10
+ test_size: 0.15
11
+ vertical_shift: -117
12
+ horizontal_shift: -25
13
+ crop:
14
+ vertical:
15
+ - 0
16
+ - 525
17
+ horizontal:
18
+ - 265
19
+ - 695
20
+ torch: true
21
+ torch_device: cuda
22
+ test_idx:
23
+ - 0
24
+ - 1
25
+ - 2
26
+ - 3
27
+ - 4
28
+ eval_disp_idx:
29
+ - 0
30
+ - 1
31
+ - 2
32
+ - 3
33
+ - 4
34
+ display:
35
+ plot: true
36
+ gamma: null
37
+ save: true
38
+ reconstruction:
39
+ method: unrolled_admm
40
+ skip_unrolled: true
41
+ unrolled_fista:
42
+ n_iter: 20
43
+ tk: 1
44
+ learn_tk: true
45
+ unrolled_admm:
46
+ n_iter: 10
47
+ mu1: 0.0001
48
+ mu2: 0.0001
49
+ mu3: 0.0001
50
+ tau: 0.0002
51
+ pre_process:
52
+ network: null
53
+ depth: 2
54
+ nc: null
55
+ post_process:
56
+ network: UnetRes
57
+ depth: 4
58
+ nc:
59
+ - 32
60
+ - 64
61
+ - 128
62
+ - 256
63
+ trainable_mask:
64
+ mask_type: null
65
+ initial_value: psf
66
+ grayscale: false
67
+ mask_lr: 0.001
68
+ L1_strength: 1.0
69
+ target: object_plane
70
+ simulation:
71
+ grayscale: false
72
+ output_dim: null
73
+ object_height: 0.33
74
+ flip: true
75
+ random_shift: false
76
+ random_vflip: false
77
+ random_hflip: false
78
+ random_rotate: false
79
+ scene2mask: 0.25
80
+ mask2sensor: 0.002
81
+ sensor: rpi_hq
82
+ snr_db: null
83
+ downsample: null
84
+ quantize: false
85
+ max_val: 255
86
+ training:
87
+ batch_size: 4
88
+ epoch: 25
89
+ eval_batch_size: 16
90
+ metric_for_best_model: null
91
+ save_every: null
92
+ skip_NAN: true
93
+ crop_preloss: true
94
+ optimizer:
95
+ type: Adam
96
+ lr: 0.0001
97
+ slow_start: false
98
+ step: false
99
+ gamma: 0.1
100
+ loss: l2
101
+ lpips: 1.0
.hydra/hydra.yaml ADDED
@@ -0,0 +1,160 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ hydra:
2
+ run:
3
+ dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}
4
+ sweep:
5
+ dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
6
+ subdir: ${hydra.job.num}
7
+ launcher:
8
+ _target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
9
+ sweeper:
10
+ _target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
11
+ max_batch_size: null
12
+ params: null
13
+ help:
14
+ app_name: ${hydra.job.name}
15
+ header: '${hydra.help.app_name} is powered by Hydra.
16
+
17
+ '
18
+ footer: 'Powered by Hydra (https://hydra.cc)
19
+
20
+ Use --hydra-help to view Hydra specific help
21
+
22
+ '
23
+ template: '${hydra.help.header}
24
+
25
+ == Configuration groups ==
26
+
27
+ Compose your configuration from those groups (group=option)
28
+
29
+
30
+ $APP_CONFIG_GROUPS
31
+
32
+
33
+ == Config ==
34
+
35
+ Override anything in the config (foo.bar=value)
36
+
37
+
38
+ $CONFIG
39
+
40
+
41
+ ${hydra.help.footer}
42
+
43
+ '
44
+ hydra_help:
45
+ template: 'Hydra (${hydra.runtime.version})
46
+
47
+ See https://hydra.cc for more info.
48
+
49
+
50
+ == Flags ==
51
+
52
+ $FLAGS_HELP
53
+
54
+
55
+ == Configuration groups ==
56
+
57
+ Compose your configuration from those groups (For example, append hydra/job_logging=disabled
58
+ to command line)
59
+
60
+
61
+ $HYDRA_CONFIG_GROUPS
62
+
63
+
64
+ Use ''--cfg hydra'' to Show the Hydra config.
65
+
66
+ '
67
+ hydra_help: ???
68
+ hydra_logging:
69
+ version: 1
70
+ formatters:
71
+ simple:
72
+ format: '[%(asctime)s][HYDRA] %(message)s'
73
+ handlers:
74
+ console:
75
+ class: logging.StreamHandler
76
+ formatter: simple
77
+ stream: ext://sys.stdout
78
+ root:
79
+ level: INFO
80
+ handlers:
81
+ - console
82
+ loggers:
83
+ logging_example:
84
+ level: DEBUG
85
+ disable_existing_loggers: false
86
+ job_logging:
87
+ version: 1
88
+ formatters:
89
+ simple:
90
+ format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
91
+ handlers:
92
+ console:
93
+ class: logging.StreamHandler
94
+ formatter: simple
95
+ stream: ext://sys.stdout
96
+ file:
97
+ class: logging.FileHandler
98
+ formatter: simple
99
+ filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
100
+ root:
101
+ level: INFO
102
+ handlers:
103
+ - console
104
+ - file
105
+ disable_existing_loggers: false
106
+ env: {}
107
+ mode: RUN
108
+ searchpath: []
109
+ callbacks: {}
110
+ output_subdir: .hydra
111
+ overrides:
112
+ hydra:
113
+ - hydra.mode=RUN
114
+ task:
115
+ - files.downsample=2
116
+ - training.batch_size=4
117
+ - reconstruction.post_process.network=UnetRes
118
+ - reconstruction.post_process.nc=[32,64,128,256]
119
+ - reconstruction.post_process.depth=4
120
+ - reconstruction.skip_unrolled=True
121
+ job:
122
+ name: train_unrolled
123
+ chdir: true
124
+ override_dirname: files.downsample=2,reconstruction.post_process.depth=4,reconstruction.post_process.nc=[32,64,128,256],reconstruction.post_process.network=UnetRes,reconstruction.skip_unrolled=True,training.batch_size=4
125
+ id: ???
126
+ num: ???
127
+ config_name: train_celeba_digicam
128
+ env_set: {}
129
+ env_copy: []
130
+ config:
131
+ override_dirname:
132
+ kv_sep: '='
133
+ item_sep: ','
134
+ exclude_keys: []
135
+ runtime:
136
+ version: 1.3.2
137
+ version_base: '1.3'
138
+ cwd: /home/bezzam/LenslessPiCam
139
+ config_sources:
140
+ - path: hydra.conf
141
+ schema: pkg
142
+ provider: hydra
143
+ - path: /home/bezzam/LenslessPiCam/configs
144
+ schema: file
145
+ provider: main
146
+ - path: ''
147
+ schema: structured
148
+ provider: schema
149
+ output_dir: /home/bezzam/LenslessPiCam/outputs/2023-10-29/12-47-05
150
+ choices:
151
+ hydra/env: default
152
+ hydra/callbacks: null
153
+ hydra/job_logging: default
154
+ hydra/hydra_logging: default
155
+ hydra/hydra_help: default
156
+ hydra/help: default
157
+ hydra/sweeper: basic
158
+ hydra/launcher: basic
159
+ hydra/output: default
160
+ verbose: false
.hydra/overrides.yaml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ - files.downsample=2
2
+ - training.batch_size=4
3
+ - reconstruction.post_process.network=UnetRes
4
+ - reconstruction.post_process.nc=[32,64,128,256]
5
+ - reconstruction.post_process.depth=4
6
+ - reconstruction.skip_unrolled=True
eval_recon/0/0.png ADDED
eval_recon/0/1.png ADDED
eval_recon/0/2.png ADDED
eval_recon/0/3.png ADDED
eval_recon/0/4.png ADDED
eval_recon/1/0.png ADDED
eval_recon/1/1.png ADDED
eval_recon/1/2.png ADDED
eval_recon/1/3.png ADDED
eval_recon/1/4.png ADDED
eval_recon/10/0.png ADDED
eval_recon/10/1.png ADDED
eval_recon/10/2.png ADDED
eval_recon/10/3.png ADDED
eval_recon/10/4.png ADDED
eval_recon/11/0.png ADDED
eval_recon/11/1.png ADDED
eval_recon/11/2.png ADDED
eval_recon/11/3.png ADDED
eval_recon/11/4.png ADDED
eval_recon/12/0.png ADDED
eval_recon/12/1.png ADDED
eval_recon/12/2.png ADDED
eval_recon/12/3.png ADDED
eval_recon/12/4.png ADDED
eval_recon/13/0.png ADDED
eval_recon/13/1.png ADDED
eval_recon/13/2.png ADDED
eval_recon/13/3.png ADDED
eval_recon/13/4.png ADDED
eval_recon/14/0.png ADDED
eval_recon/14/1.png ADDED
eval_recon/14/2.png ADDED
eval_recon/14/3.png ADDED
eval_recon/14/4.png ADDED
eval_recon/15/0.png ADDED
eval_recon/15/1.png ADDED
eval_recon/15/2.png ADDED
eval_recon/15/3.png ADDED
eval_recon/15/4.png ADDED
eval_recon/16/0.png ADDED
eval_recon/16/1.png ADDED
eval_recon/16/2.png ADDED
eval_recon/16/3.png ADDED
eval_recon/16/4.png ADDED
eval_recon/17/0.png ADDED