Hugging Face's logo

wilecoyote
/
raw_denoise

Image-to-Image
denoising
raw
demosaicing
Model card Files
xet
 Community
wilecoyote commited on
Commit
249c3b1
·
verified ·
1 Parent(s): 6c7c6d7

Create general_models/bayer/nafnet_like_paper_window_256_conv_encoder/dists_l1_same_scale_general_model_combined_gain_readnoise_sampling_10_samples/model_info.json

Browse files
Files changed (1) hide show
  1. general_models/bayer/nafnet_like_paper_window_256_conv_encoder/dists_l1_same_scale_general_model_combined_gain_readnoise_sampling_10_samples/model_info.json +468 -0
general_models/bayer/nafnet_like_paper_window_256_conv_encoder/dists_l1_same_scale_general_model_combined_gain_readnoise_sampling_10_samples/model_info.json ADDED
@@ -0,0 +1,468 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "denoise_model_config": {
3
+ "name": "nafnet_like_paper_window_256_conv_encoder",
4
+ "noise_predictor_model_type": "nafnet",
5
+ "noise_predictor_model_params": {
6
+ "width": 64,
7
+ "enc_blk_nums": [
8
+ 2,
9
+ 2,
10
+ 4,
11
+ 8
12
+ ],
13
+ "middle_blk_num": 12,
14
+ "dec_blk_nums": [
15
+ 2,
16
+ 2,
17
+ 2,
18
+ 2
19
+ ]
20
+ },
21
+ "settings_encoder_model_type": "conv",
22
+ "settings_encoder_n_output_channels": 2,
23
+ "settings_encoder_model_params": {
24
+ "n_units_layers_dense": [
25
+ 32,
26
+ 64,
27
+ 128
28
+ ],
29
+ "size_encoded": 16,
30
+ "n_channels_encoded": 2,
31
+ "n_channels_decoder_blocks": 8,
32
+ "use_tanh_output": false
33
+ },
34
+ "use_downscaled_demosaiced_context_as_model_input": true,
35
+ "use_anscombe_transform": true,
36
+ "use_channel_indicators_as_model_input": true,
37
+ "use_encoded_settings_as_model_input": true,
38
+ "use_demosaiced_image_as_model_input": true,
39
+ "use_mosaiced_image_as_model_input": true,
40
+ "border_size": 8,
41
+ "window_size_input": 256,
42
+ "predict_noise": true
43
+ },
44
+ "training_config": {
45
+ "name": "dists_l1_same_scale_general_model_combined_gain_readnoise_sampling_10_samples",
46
+ "optimizer": "Adam",
47
+ "optimizer_kwargs": {
48
+ "betas": [
49
+ 0.9,
50
+ 0.999
51
+ ],
52
+ "weight_decay": 0
53
+ },
54
+ "learning_rate_start": 0.0001,
55
+ "learning_rate_scheduler": "CosineAnnealingWarmRestarts",
56
+ "learning_rate_scheduler_kwargs": {
57
+ "T_0": 100,
58
+ "eta_min": 0.00001
59
+ },
60
+ "loss_function_kwargs": {
61
+ "DISTS": {
62
+ "weight_l1_loss": 0,
63
+ "weight_chroma_loss": 0,
64
+ "do_shifting": false,
65
+ "max_shift": 4,
66
+ "normalize_channels": true,
67
+ "normalize_values": true,
68
+ "n_stops_exposure_correction": [
69
+ 0.0
70
+ ],
71
+ "max_scale_factor_normalize_values": 10.0,
72
+ "max_scale_factor_normalize_channels": 10.0,
73
+ "resize_to_256": true
74
+ },
75
+ "L1": {
76
+ "normalize_channels": true,
77
+ "normalize_values": true,
78
+ "max_scale_factor_normalize_values": 10.0,
79
+ "max_scale_factor_normalize_channels": 10.0,
80
+ "value_norm_range": "01"
81
+ }
82
+ },
83
+ "n_epochs": 100,
84
+ "batch_size": 32,
85
+ "window_stride": 64,
86
+ "prob_wb_modification_coeffs": 0.5,
87
+ "range_exposure": [
88
+ -5.0,
89
+ 0.0
90
+ ],
91
+ "names_image_dirs_validation": [
92
+ "_DSF2766",
93
+ "_DSF1515",
94
+ "Sigma56mm-Z-C1-21",
95
+ "_DSF1969",
96
+ "DSCF5031",
97
+ "DSCF2259",
98
+ "DSCF3573",
99
+ "DSCF3531",
100
+ "DSCF3271",
101
+ "_DSF2329",
102
+ "DSC00713",
103
+ "Sigma56mm-Z-C1-02",
104
+ "fujifilm_x_t3_90",
105
+ "DSC01016",
106
+ "DSCF2282",
107
+ "_DSF1451",
108
+ "DSC01334",
109
+ "459A0267",
110
+ "DPR22206",
111
+ "459A9991",
112
+ "P1015382",
113
+ "DSCF0016",
114
+ "DSCF3516",
115
+ "_DSF1487",
116
+ "DSCF2008",
117
+ "DSCF2039",
118
+ "DSCF0521",
119
+ "DSCF3768",
120
+ "DSC00190",
121
+ "DSC00562",
122
+ "DSC00801",
123
+ "DSC01342",
124
+ "_DSF1968",
125
+ "DSCF2060",
126
+ "DSC00628",
127
+ "DSC01205",
128
+ "0E3A8989",
129
+ "DSC00916",
130
+ "fujifilm_x_t3_21",
131
+ "DSC08665",
132
+ "fujifilm_x_t3_01",
133
+ "DSCF0526",
134
+ "DSCF0256",
135
+ "P1025458",
136
+ "Sigma56mm-Z-C1-24",
137
+ "DSCF2622",
138
+ "_1100025",
139
+ "DSCF0847",
140
+ "IMG_0540",
141
+ "_DSF2233",
142
+ "DSC08805",
143
+ "DSCF2225",
144
+ "DSCF3678",
145
+ "_DSF1431",
146
+ "fujifilm_x_t3_82",
147
+ "Sigma56mm-Z-C1-35",
148
+ "Sigma56mm-Z-C1-01",
149
+ "DSC00766",
150
+ "DSC00001",
151
+ "DSCF3192",
152
+ "DSC08715",
153
+ "_1100477",
154
+ "DSCF0309",
155
+ "_DSF2421",
156
+ "DSC01182 (2.8 jordan)",
157
+ "_DSF1705",
158
+ "fujifilm_x_t3_48",
159
+ "DSCF0126",
160
+ "IMG_0440",
161
+ "459A0050",
162
+ "DSCF0026_1",
163
+ "DSCF0057",
164
+ "_DSF1698",
165
+ "DSCF0011",
166
+ "_DSF1644",
167
+ "DSCF0849",
168
+ "Sigma56mm-Z-C1-39",
169
+ "DSCF3584",
170
+ "_DSF2467",
171
+ "DSC00333",
172
+ "fujifilm_x_t3_79",
173
+ "DSCF0058",
174
+ "DSCF3814",
175
+ "DSCF3763",
176
+ "DSCF3559",
177
+ "_1100512",
178
+ "DSCF3510",
179
+ "fujifilm_x_t3_26",
180
+ "DSCF3167",
181
+ "_1100122",
182
+ "DSCF0002",
183
+ "_DSF2293",
184
+ "_DSF2287",
185
+ "_DSF1508",
186
+ "DSCF3267",
187
+ "_DSF2771",
188
+ "P1020178",
189
+ "Sigma56mm-Z-C1-14",
190
+ "fujifilm_x_t3_51",
191
+ "0E3A8794"
192
+ ],
193
+ "order_channel_indices_demosaiced_image": [
194
+ 0,
195
+ 1,
196
+ 2
197
+ ],
198
+ "relative_std_conversion_factor_measurement_error": 0.0,
199
+ "relative_std_read_noise_random_shift": 0.05,
200
+ "relative_std_read_noise_random_scale": 0.0,
201
+ "ratio_generated_patterns": 0.2,
202
+ "loss_input_values_transform": "anscombe_transformed_unnormalized",
203
+ "probability_anti_aliasing_of_random_patterns": 0.5,
204
+ "read_noise_sampling_mode": "window",
205
+ "de_discretize_read_noise": true,
206
+ "name_read_noise_distribution_de_discretize": "sinh_arcsinh",
207
+ "name_read_noise_distribution_sample": "student_t",
208
+ "n_pixels_sample_read_noise": 1000000,
209
+ "n_raw_patterns_window_stride_read_noise": 10,
210
+ "n_samples_de_discretize_read_noise": 10,
211
+ "n_samples": 1,
212
+ "cache_sampled_read_noise_windows": false,
213
+ "multi_loss_combiner": "exact_config",
214
+ "use_combined_read_noise_std_and_gain_model": true,
215
+ "type_config": "general_model",
216
+ "gamma_range_extension_factor": 1.5,
217
+ "prob_quantize": 0.5
218
+ },
219
+ "global_camera_profile": {
220
+ "raw_pattern": [
221
+ [
222
+ 0,
223
+ 1
224
+ ],
225
+ [
226
+ 1,
227
+ 2
228
+ ]
229
+ ],
230
+ "gamma_range": [
231
+ 4.192708333333334,
232
+ 1087.4074074074074
233
+ ],
234
+ "camera_models": [
235
+ {
236
+ "maker": "SONY",
237
+ "model": "ILCE-7M4"
238
+ },
239
+ {
240
+ "maker": "SONY",
241
+ "model": "NEX-5N"
242
+ },
243
+ {
244
+ "maker": "SONY",
245
+ "model": "DSC-RX100M3"
246
+ }
247
+ ],
248
+ "model_parameters_read_noise_stds_colors": {
249
+ "parameters_polynomial": [
250
+ [
251
+ -0.39264822006225586
252
+ ],
253
+ [
254
+ 1.1299388408660889
255
+ ],
256
+ [
257
+ -0.0690825954079628
258
+ ],
259
+ [
260
+ 0.3892571032047272
261
+ ]
262
+ ],
263
+ "std": [
264
+ 0.009999999776482582
265
+ ],
266
+ "unconstrained_corr_matrix": [
267
+ [
268
+ 0.4511893093585968,
269
+ 0.005226525943726301,
270
+ 0.004171822685748339
271
+ ],
272
+ [
273
+ 1.5619579553604126,
274
+ -2.763054370880127,
275
+ 0.003968189936131239
276
+ ],
277
+ [
278
+ 1.6363657712936401,
279
+ 0.007545437663793564,
280
+ -2.426279067993164
281
+ ]
282
+ ],
283
+ "intercept_scale_vs_gamma": [
284
+ -0.028559770435094833
285
+ ],
286
+ "slope_scale_vs_gamma": [
287
+ 0.08118516951799393
288
+ ],
289
+ "std_gamma": 288.1870422363281,
290
+ "std_output": [
291
+ [
292
+ 0.002197723602876067,
293
+ 0.002226460026577115,
294
+ 0.0021788605954498053
295
+ ]
296
+ ],
297
+ "min_output_factor": 0.5,
298
+ "min_output": [
299
+ [
300
+ 0.00009186257375404239,
301
+ 0.00009265590779250488,
302
+ 0.00008408483699895442
303
+ ]
304
+ ]
305
+ },
306
+ "model_parameters_gains_colors": {
307
+ "parameters_polynomial": [
308
+ [
309
+ -0.6041020750999451
310
+ ],
311
+ [
312
+ 1.3148465156555176
313
+ ],
314
+ [
315
+ -0.07865296304225922
316
+ ],
317
+ [
318
+ 0.47162675857543945
319
+ ]
320
+ ],
321
+ "std": [
322
+ 0.009999999776482582
323
+ ],
324
+ "unconstrained_corr_matrix": [
325
+ [
326
+ -0.6111693382263184,
327
+ 0.004225569311529398,
328
+ -0.00838350411504507
329
+ ],
330
+ [
331
+ 0.4591483771800995,
332
+ -1.9379860162734985,
333
+ -0.004047973547130823
334
+ ],
335
+ [
336
+ 0.5257622599601746,
337
+ 0.1745738834142685,
338
+ -0.692702054977417
339
+ ]
340
+ ],
341
+ "intercept_scale_vs_gamma": [
342
+ -0.046116143465042114
343
+ ],
344
+ "slope_scale_vs_gamma": [
345
+ 0.09076843410730362
346
+ ],
347
+ "std_gamma": 288.1870422363281,
348
+ "std_output": [
349
+ [
350
+ 0.0011818252969533205,
351
+ 0.001211264985613525,
352
+ 0.0011370822321623564
353
+ ]
354
+ ],
355
+ "min_output_factor": 0.5,
356
+ "min_output": [
357
+ [
358
+ 0.000020191202565911226,
359
+ 0.000020973444406990893,
360
+ 0.000018127228031517006
361
+ ]
362
+ ]
363
+ },
364
+ "model_parameters_stds_read_noise_and_gains_colors": {
365
+ "parameters_polynomial": [
366
+ [
367
+ -0.3802710175514221,
368
+ -0.6156619191169739
369
+ ],
370
+ [
371
+ 1.1284230947494507,
372
+ 1.3155611753463745
373
+ ],
374
+ [
375
+ -0.0690828487277031,
376
+ -0.07887454330921173
377
+ ],
378
+ [
379
+ 0.3782936930656433,
380
+ 0.47987788915634155
381
+ ]
382
+ ],
383
+ "std": [
384
+ 0.009999999776482582
385
+ ],
386
+ "unconstrained_corr_matrix": [
387
+ [
388
+ 0.7508730888366699,
389
+ 0.0065185073763132095,
390
+ -0.004930655471980572,
391
+ 0.01181046012789011,
392
+ 0.0037522318307310343,
393
+ 0.005557376425713301
394
+ ],
395
+ [
396
+ 2.106855630874634,
397
+ -2.4363467693328857,
398
+ 0.0007946193218231201,
399
+ -0.01619265414774418,
400
+ -0.015816491097211838,
401
+ 0.014647204428911209
402
+ ],
403
+ [
404
+ 2.2110371589660645,
405
+ 0.009627812542021275,
406
+ -2.171307325363159,
407
+ 0.002969216089695692,
408
+ 0.019714046269655228,
409
+ 0.011513433419167995
410
+ ],
411
+ [
412
+ 0.12182650715112686,
413
+ 0.2866002023220062,
414
+ -0.17357172071933746,
415
+ -0.9270530343055725,
416
+ -0.0025384484324604273,
417
+ 0.008513505570590496
418
+ ],
419
+ [
420
+ 0.08325459063053131,
421
+ 0.2331448495388031,
422
+ -0.09681519865989685,
423
+ 0.3585951626300812,
424
+ -2.0144739151000977,
425
+ -0.012930363416671753
426
+ ],
427
+ [
428
+ 0.5369732975959778,
429
+ 0.17834579944610596,
430
+ 0.07175642997026443,
431
+ 0.4162815511226654,
432
+ 0.14157187938690186,
433
+ -1.9565629959106445
434
+ ]
435
+ ],
436
+ "intercept_scale_vs_gamma": [
437
+ -0.03131347894668579,
438
+ -0.043236445635557175
439
+ ],
440
+ "slope_scale_vs_gamma": [
441
+ 0.05940856412053108,
442
+ 0.07936418801546097
443
+ ],
444
+ "std_gamma": 288.1870422363281,
445
+ "std_output": [
446
+ [
447
+ 0.002197723602876067,
448
+ 0.002226460026577115,
449
+ 0.0021788605954498053,
450
+ 0.0011818252969533205,
451
+ 0.001211264985613525,
452
+ 0.0011370822321623564
453
+ ]
454
+ ],
455
+ "min_output_factor": 0.5,
456
+ "min_output": [
457
+ [
458
+ 0.00009186257375404239,
459
+ 0.00009265590779250488,
460
+ 0.00008408483699895442,
461
+ 0.000020191202565911226,
462
+ 0.000020973444406990893,
463
+ 0.000018127228031517006
464
+ ]
465
+ ]
466
+ }
467
+ }
468
+ }