| { | |
| "backend": { | |
| "id": "torch", | |
| "device": "none" | |
| }, | |
| "dataset_training": { | |
| "dataset_list": [ | |
| { | |
| "image": { | |
| "id": null, | |
| "dataset": "chikusei", | |
| "dataset_id": 0 | |
| }, | |
| "preprocessing": { | |
| "crop_row": [ | |
| 64, | |
| 1664 | |
| ], | |
| "crop_col": [ | |
| 64, | |
| 2240 | |
| ], | |
| "crop_chn": [ | |
| 8, | |
| 72, | |
| 4 | |
| ], | |
| "crop_acq": null, | |
| "select_chn": null, | |
| "normalize_intensity": true, | |
| "equalize_dynamic": true, | |
| "dynamic_range": [ | |
| 0.05, | |
| 0.95 | |
| ] | |
| } | |
| } | |
| ], | |
| "patch_size": 64, | |
| "stride": 16, | |
| "horizontal": true, | |
| "vertical": true, | |
| "rotation": true | |
| }, | |
| "dataset_validation": { | |
| "dataset_list": [ | |
| { | |
| "image": { | |
| "id": null, | |
| "dataset": "chikusei", | |
| "dataset_id": 0 | |
| }, | |
| "preprocessing": { | |
| "crop_row": [ | |
| 1664, | |
| 2048 | |
| ], | |
| "crop_col": [ | |
| 64, | |
| 2240 | |
| ], | |
| "crop_chn": [ | |
| 8, | |
| 72, | |
| 4 | |
| ], | |
| "crop_acq": null, | |
| "select_chn": null, | |
| "normalize_intensity": true, | |
| "equalize_dynamic": true, | |
| "dynamic_range": [ | |
| 0.05, | |
| 0.95 | |
| ] | |
| } | |
| } | |
| ], | |
| "patch_size": 64, | |
| "stride": 64, | |
| "horizontal": false, | |
| "vertical": false, | |
| "rotation": false | |
| }, | |
| "model": { | |
| "linear_id": "deblurring", | |
| "kernel_id": 0, | |
| "color_filter_id": null, | |
| "spectral_responses_id": null, | |
| "spectral_responses_type": "multispectral", | |
| "scale_ratio": null, | |
| "lagrange_multipliers": [] | |
| }, | |
| "noise": { | |
| "id": "gaussian", | |
| "snr": 25.0, | |
| "mean_scaled_snr": true | |
| }, | |
| "processing": { | |
| "data_functional": "squared_l2", | |
| "priors": [ | |
| { | |
| "id": "pnp", | |
| "unfolded_stages": 4, | |
| "denoiser_deep": { | |
| "id": "unet", | |
| "in_channels": 16 | |
| }, | |
| "regularization_parameter": 1.0, | |
| "noise_level": 0.03 | |
| } | |
| ], | |
| "solver": { | |
| "id": "admm", | |
| "penalization_parameter_list": [ | |
| 0.1 | |
| ], | |
| "penalization_parameter": 0.1, | |
| "penalization_parameter_dual": 0.1, | |
| "relaxation_parameter": 0.1, | |
| "sub_problem_solver": "auto", | |
| "max_iterations": 1, | |
| "verbose": true, | |
| "g_first": false, | |
| "early_stop": false, | |
| "nan_stop": false | |
| } | |
| }, | |
| "dataloader_options": { | |
| "shuffle_train": true, | |
| "shuffle_validation": false, | |
| "num_workers": 4, | |
| "batch_size": 48 | |
| }, | |
| "trainable_parameters": { | |
| "regularization_parameter": false, | |
| "noise_level": false, | |
| "relaxation_parameter": true, | |
| "penalization_parameter": true, | |
| "penalization_parameter_dual": false | |
| }, | |
| "training_options": { | |
| "optimizer": "adam", | |
| "optimizer_momentum": 0.9, | |
| "learning_rate": 0.01, | |
| "loss_validation": "mse", | |
| "loss_test": [ | |
| "mse", | |
| "ssim", | |
| "psnr" | |
| ], | |
| "scheduler": { | |
| "id": "reduce_on_plateau", | |
| "max_iterations": 50, | |
| "reduction_factor": 0.1, | |
| "patience": 10, | |
| "tolerance": 1e-6 | |
| } | |
| }, | |
| "training_logger": { | |
| "logger": "csv", | |
| "maximum_time": "00:00:40:00", | |
| "early_stopping_delta": 1e-6, | |
| "early_stopping_patience": 20, | |
| "tqdm_refresh_rate": 10 | |
| } | |
| } | |