Daniele Picone commited on
Commit
f5b9e10
·
1 Parent(s): abfd636

AddedAdded Harvard dataset

Browse files
20260129_131831_409082/lightning_logs/version_0/checkpoints/best.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3606c784f409a5907faf9baee9a33f3e4695df61d335c6a3bc0b93db653cf10
3
+ size 130616004
20260129_131831_409082/lightning_logs/version_0/checkpoints/last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6339a74c4814780ece8d9d4a23bc8f8c003430a3c5feba68b724df4b2f4e4ca9
3
+ size 130616004
20260129_131831_409082/lightning_logs/version_0/metrics.csv ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ epoch,step,train_loss_epoch,val_loss
2
+ 0,0,,0.10943936556577682
3
+ 0,0,0.2632017731666565,
4
+ 1,1,,0.06855326145887375
5
+ 1,1,0.16630996763706207,
6
+ 2,2,,0.04021093621850014
7
+ 2,2,0.10109398514032364,
8
+ 3,3,,0.021488085389137268
9
+ 3,3,0.056322336196899414,
10
+ 4,4,,0.010715650394558907
11
+ 4,4,0.025746257975697517,
12
+ 5,5,,0.005771419499069452
13
+ 5,5,0.008420568890869617,
14
+ 6,6,,0.0043568979017436504
15
+ 6,6,0.0016469513066112995,
16
+ 7,7,,0.00401181960478425
17
+ 7,7,0.0020498863887041807,
18
+ 8,8,,0.003853239817544818
19
+ 8,8,0.00230778637342155,
20
+ 9,9,,0.003786688204854727
21
+ 9,9,0.0011925420258194208,
22
+ 10,10,,0.0037957774475216866
23
+ 10,10,0.002207427518442273,
24
+ 11,11,,0.003868784988299012
25
+ 11,11,0.002403022488579154,
26
+ 12,12,,0.0039802780374884605
27
+ 12,12,0.0025380149018019438,
28
+ 13,13,,0.004098051227629185
29
+ 13,13,0.002473702421411872,
30
+ 14,14,,0.0041653974913060665
31
+ 14,14,0.002372465329244733,
32
+ 15,15,,0.0041834707371890545
33
+ 15,15,0.0024568731896579266,
34
+ 16,16,,0.004181369673460722
35
+ 16,16,0.002283988520503044,
36
+ 17,17,,0.004212250467389822
37
+ 17,17,0.002088770270347595,
38
+ 18,18,,0.004253508523106575
39
+ 18,18,0.002092937706038356,
40
+ 19,19,,0.004281893838196993
41
+ 19,19,0.002377263270318508,
42
+ 20,20,,0.004260672722011805
43
+ 20,20,0.0021507274359464645,
44
+ 21,21,,0.004253780003637075
45
+ 21,21,0.0014092191122472286,
46
+ 22,22,,0.00423910329118371
47
+ 22,22,0.0023076448123902082,
48
+ 23,23,,0.0042198533192276955
49
+ 23,23,0.0023019283544272184,
50
+ 24,24,,0.004196014255285263
51
+ 24,24,0.0022932959254831076,
52
+ 25,25,,0.004166525788605213
53
+ 25,25,0.0013789883814752102,
54
+ 26,26,,0.004137144889682531
55
+ 26,26,0.0021267386619001627,
56
+ 27,27,,0.004105245228856802
57
+ 27,27,0.0022559938952326775,
58
+ 28,28,,0.004068923182785511
59
+ 28,28,0.0023588647600263357,
60
+ 29,29,,0.004033316858112812
61
+ 29,29,0.0023459908552467823,
20260129_131831_409082/manifest.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "version": "0.10.0",
3
+ "options": "options.json",
4
+ "denoiser": "lightning_logs/version_0/checkpoints/best.ckpt"
5
+ }
20260129_131831_409082/options.json ADDED
@@ -0,0 +1,243 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "backend": {
3
+ "id": "local",
4
+ "device": "none"
5
+ },
6
+ "dataset_train": {
7
+ "dataset_list": [
8
+ {
9
+ "image": {
10
+ "id": null,
11
+ "dataset": "harvard",
12
+ "dataset_id": 0
13
+ },
14
+ "preprocessing": {
15
+ "crop_row": null,
16
+ "crop_col": null,
17
+ "crop_chn": null,
18
+ "crop_acq": [
19
+ 0,
20
+ 14
21
+ ],
22
+ "select_chn": null,
23
+ "select_acq": null,
24
+ "normalize_intensity": false,
25
+ "equalize_dynamic": true,
26
+ "equalize_per_band": true,
27
+ "dynamic_range": [
28
+ 0.01,
29
+ 0.99
30
+ ],
31
+ "white_reference_pixel": null,
32
+ "white_reference_acquisition": null
33
+ }
34
+ },
35
+ {
36
+ "image": {
37
+ "id": null,
38
+ "dataset": "harvard",
39
+ "dataset_id": 0
40
+ },
41
+ "preprocessing": {
42
+ "crop_row": null,
43
+ "crop_col": null,
44
+ "crop_chn": null,
45
+ "crop_acq": [
46
+ 20,
47
+ 60
48
+ ],
49
+ "select_chn": null,
50
+ "select_acq": null,
51
+ "normalize_intensity": false,
52
+ "equalize_dynamic": true,
53
+ "equalize_per_band": true,
54
+ "dynamic_range": [
55
+ 0.01,
56
+ 0.99
57
+ ],
58
+ "white_reference_pixel": null,
59
+ "white_reference_acquisition": null
60
+ }
61
+ }
62
+ ],
63
+ "patch_size": 64,
64
+ "stride": 64,
65
+ "horizontal": true,
66
+ "vertical": false,
67
+ "rotation": false
68
+ },
69
+ "dataset_validation": {
70
+ "dataset_list": [
71
+ {
72
+ "image": {
73
+ "id": null,
74
+ "dataset": "harvard",
75
+ "dataset_id": 0
76
+ },
77
+ "preprocessing": {
78
+ "crop_row": null,
79
+ "crop_col": null,
80
+ "crop_chn": null,
81
+ "crop_acq": [
82
+ 14,
83
+ 17
84
+ ],
85
+ "select_chn": null,
86
+ "select_acq": null,
87
+ "normalize_intensity": false,
88
+ "equalize_dynamic": true,
89
+ "equalize_per_band": true,
90
+ "dynamic_range": [
91
+ 0.01,
92
+ 0.99
93
+ ],
94
+ "white_reference_pixel": null,
95
+ "white_reference_acquisition": null
96
+ }
97
+ },
98
+ {
99
+ "image": {
100
+ "id": null,
101
+ "dataset": "harvard",
102
+ "dataset_id": 0
103
+ },
104
+ "preprocessing": {
105
+ "crop_row": null,
106
+ "crop_col": null,
107
+ "crop_chn": null,
108
+ "crop_acq": [
109
+ 60,
110
+ 69
111
+ ],
112
+ "select_chn": null,
113
+ "select_acq": null,
114
+ "normalize_intensity": false,
115
+ "equalize_dynamic": true,
116
+ "equalize_per_band": true,
117
+ "dynamic_range": [
118
+ 0.01,
119
+ 0.99
120
+ ],
121
+ "white_reference_pixel": null,
122
+ "white_reference_acquisition": null
123
+ }
124
+ }
125
+ ],
126
+ "patch_size": 64,
127
+ "stride": 64,
128
+ "horizontal": false,
129
+ "vertical": false,
130
+ "rotation": false
131
+ },
132
+ "noise": [
133
+ {
134
+ "id": "gaussian",
135
+ "snr": null,
136
+ "mean_scaled_snr": true,
137
+ "noise_level": 0.05,
138
+ "apply_before_degradation": false
139
+ },
140
+ {
141
+ "id": "gaussian",
142
+ "snr": null,
143
+ "mean_scaled_snr": true,
144
+ "noise_level": 0.1,
145
+ "apply_before_degradation": false
146
+ },
147
+ {
148
+ "id": "gaussian",
149
+ "snr": null,
150
+ "mean_scaled_snr": true,
151
+ "noise_level": 0.15,
152
+ "apply_before_degradation": false
153
+ },
154
+ {
155
+ "id": "gaussian",
156
+ "snr": null,
157
+ "mean_scaled_snr": true,
158
+ "noise_level": 0.2,
159
+ "apply_before_degradation": false
160
+ }
161
+ ],
162
+ "denoiser_options": {
163
+ "id": "drunet_color",
164
+ "in_channels": null,
165
+ "noise_level": null,
166
+ "layers": null,
167
+ "layer_filter_list": [
168
+ 64,
169
+ 128,
170
+ 256
171
+ ],
172
+ "layer_filters": null,
173
+ "stages_per_layer": 2,
174
+ "activator": "relu",
175
+ "downsample": "strided_convolution",
176
+ "upsample": "convolution_transpose",
177
+ "bias": null,
178
+ "pretrained": true,
179
+ "manifest_path": null
180
+ },
181
+ "dataloader_options": {
182
+ "shuffle_train": true,
183
+ "shuffle_validation": false,
184
+ "num_workers": 32,
185
+ "batch_size": 128,
186
+ "interim_dump": true
187
+ },
188
+ "logging_options": {
189
+ "logger": "csv",
190
+ "maximum_time": "00:48:00:00",
191
+ "max_epochs": null,
192
+ "early_stopping_delta": 1e-06,
193
+ "early_stopping_patience": 20,
194
+ "tqdm_refresh_rate": 50,
195
+ "batch_finder": false
196
+ },
197
+ "training_options": {
198
+ "optimizer": "adam",
199
+ "optimizer_momentum": 0.9,
200
+ "learning_rate": 0.01,
201
+ "loss_validation": "mse",
202
+ "loss_test": [
203
+ "mse",
204
+ "sam",
205
+ "ssim",
206
+ "psnr"
207
+ ],
208
+ "scheduler": {
209
+ "id": "reduce_on_plateau",
210
+ "max_iterations": 50,
211
+ "reduction_factor": 0.1,
212
+ "patience": 10,
213
+ "tolerance": 1e-06
214
+ }
215
+ },
216
+ "adaptor_options": {
217
+ "enable": true,
218
+ "train_denoiser": false,
219
+ "grouper": {
220
+ "group_id": "sequential",
221
+ "config": {},
222
+ "batched": true
223
+ },
224
+ "projection": {
225
+ "projection_id": "channel_linear_transform",
226
+ "projection_rank": 31,
227
+ "config": {
228
+ "regularizer": "orthogonal",
229
+ "temperature": 1.0,
230
+ "pinv_lambda": 0.0001,
231
+ "group_size": 1
232
+ }
233
+ },
234
+ "skip": {
235
+ "skip_id": "none",
236
+ "config": null
237
+ },
238
+ "hook": {
239
+ "hook_id": "none",
240
+ "config": null
241
+ }
242
+ }
243
+ }
README.md CHANGED
@@ -27,6 +27,7 @@ In particular the weights contained in this folder are associated to the followi
27
  - `20251105_024251_967712` - `cave_projection_matrix_orthogonal_compressed`: Linear channel projection matrix to 3 channels. Matrix is constrained to be orthogonal.
28
  - `20251104_200826_195587` - `cave_projection_matrix_simplex`: Linear channel projection matrix constrained to be positive with sum-to-one condition.
29
  - `20251104_230322_748168` - `cave_projection_matrix_simplex_compressed`: Linear channel projection matrix to 3 channels. Matrix is constrained to be positive with sum-to-one condition.
 
30
 
31
  # Credits
32
 
 
27
  - `20251105_024251_967712` - `cave_projection_matrix_orthogonal_compressed`: Linear channel projection matrix to 3 channels. Matrix is constrained to be orthogonal.
28
  - `20251104_200826_195587` - `cave_projection_matrix_simplex`: Linear channel projection matrix constrained to be positive with sum-to-one condition.
29
  - `20251104_230322_748168` - `cave_projection_matrix_simplex_compressed`: Linear channel projection matrix to 3 channels. Matrix is constrained to be positive with sum-to-one condition.
30
+ - `202601209_131831_409082` - `harvard_projection_matrix_orthogonal`: Linear channel projection, with orthogonal matrix, trained on the Harvard dataset.
31
 
32
  # Credits
33