PentHertz
/
deep-tempest
mirror of https://github.com/PentHertz/deep-tempest.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update options JSON files

This commit is contained in:
Emilio Martinez 2023-10-20 18:15:21 -03:00
parent 246e818505
commit f2d68d8994
5 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
KAIR/options/optuna_options.json
View File

@ -27,8 +27,8 @@
"train": {
"name": "train_dataset" // just name
, "dataset_type": "ffdnet" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": "trainsets/ground-truth" // path of H training dataset
, "dataroot_L": "trainsets/simulations" // path of L training dataset, if using noisy H type: null
, "dataroot_H": "path/to/train_original" // path of H training dataset
, "dataroot_L": "path/to/train_degraded" // path of L training dataset, if using noisy H type: null
, "sigma": [0, 20] // 15, 25, 50 for DnCNN | [0, 75] for FFDNet and FDnCNN
, "num_patches_per_image": 21 // number of random patches of training image
, "H_size": 256 // patch size 40 | 64 | 96 | 128 | 192
@ -39,8 +39,8 @@
, "test": {
"name": "test_dataset" // just name
, "dataset_type": "ffdnet" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": "valsets/ground-truth" // path of H testing dataset
, "dataroot_L": "valsets/simulations" // path of L testing dataset
, "dataroot_H": "path/to/val_original" // path of H testing dataset
, "dataroot_L": "path/to/val_degraded" // path of L testing dataset
, "sigma_test": 15 // 15, 25, 50 for DnCNN and ffdnet
}
}

4
KAIR/options/tempest_simulation.json
View File

@ -1,9 +1,9 @@
{
"paths": {
"folder_original_images": "web_imgs_tmp/test",
"folder_original_images": "path/to/folder",
"__comment1__": "Insert input folder (original images)",
"folder_simulated_images": "testsets/simulations",
"folder_simulated_images": "path/to/new_folder",
"__comment2__": "Insert output folder (tempest degraded images)"
},

6
KAIR/options/test_drunet.json
View File

@ -9,7 +9,7 @@
, "path": {
"root": "drunet_inference" // "denoising" | "superresolution"
, "pretrained_netG": "denoising/drunet/models/70_G.pth" // path of pretrained model, if model from scratch type: null
, "pretrained_netG": "path/to/model" // path of pretrained model, if model from scratch type: null
}
, "datasets": {
@ -28,8 +28,8 @@
, "test": {
"name": "test_dataset" // just name
, "dataset_type": "ffdnet" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": "trainsets/ground-truth" // path of H testing dataset
, "dataroot_L": "trainsets/simulations" // path of L testing dataset
, "dataroot_H": "web_subset/capturas_splitter" // path of H testing dataset
, "dataroot_L": "web_subset/capturas_splitter" // path of L testing dataset
, "sigma_test": 0 // 15, 25, 50 for DnCNN and ffdnet
}
}

12
KAIR/options/train_drunet.json
View File

@ -8,16 +8,16 @@
, "n_channels_datasetload": 3 // broadcast to image training set
, "path": {
"root": "denoising" // "denoising" | "superresolution"
, "pretrained_netG": "denoising/drunet/models/170_G.pth" // path of pretrained model, if model from scratch type: null
"root": "drunet_training" // "denoising" | "superresolution"
, "pretrained_netG": "path/to/model" // path of pretrained model, if model from scratch type: null
}
, "datasets": {
"train": {
"name": "train_dataset" // just name
, "dataset_type": "ffdnet" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": "train_val/ground-truth"// path of H training dataset
, "dataroot_L": "train_val/simulations" // path of L training dataset, if using noisy H type: null
, "dataroot_H": "path/to/train_original"// path of H training dataset
, "dataroot_L": "path/to/train_degraded" // path of L training dataset, if using noisy H type: null
, "sigma": [0, 20] // 15, 25, 50 for DnCNN | [0, 75] for FFDNet and FDnCNN
, "use_all_patches": true // use or not all image patches
, "skip_natural_patches": false// keep only non-natural image patches/text based image patches
@ -30,8 +30,8 @@
, "test": {
"name": "test_dataset" // just name
, "dataset_type": "ffdnet" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": null // path of H testing dataset
, "dataroot_L": null // path of L testing dataset
, "dataroot_H": "path/to/val_original" // path of H testing dataset
, "dataroot_L": "path/to/val_degraded" // path of L testing dataset
, "sigma_test": 15 // 15, 25, 50 for DnCNN and ffdnet
}
}

8
KAIR/options/train_drunet_finetuning.json
View File

@ -8,15 +8,15 @@
, "n_channels_datasetload": 3 // broadcast to image training set
, "path": {
"root": "max_entropy_thrs_test" // "finetuning_{border, middle}"
, "pretrained_netG": "finetuning_down2_down3_up3_up2/drunet/models/100_G.pth" // path of pretrained model, if model from scratch type: null
"root": "finetuning" // "finetuning_{border, middle}"
, "pretrained_netG": "path/to/model" // path of pretrained model, if model from scratch type: null
}
, "datasets": {
"train": {
"name": "train_dataset" // just name
, "dataset_type": "drunet_finetune" // "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": null // path of H training dataset
, "dataroot_H": "path/to/train_original" // path of H training dataset
, "dataroot_L": null // path of L training dataset, not used for finetuning
, "sigma": [0, 0] // 15, 25, 50 for DnCNN | [0, 75] for FFDNet and FDnCNN
, "use_all_patches": true // use or not all image patches
@ -30,7 +30,7 @@
, "test": {
"name": "test_dataset" // just name
, "dataset_type": "drunet_finetune" // "drunet" | "drunet_finetune" | "dncnn" | "dnpatch" for dncnn, | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
, "dataroot_H": "finetuning_images/test" // path of H testing dataset
, "dataroot_H": "path/to/val_original" // path of H testing dataset
, "dataroot_L": null // path of L testing dataset, not used for finetuning
, "sigma_test": 0 // 15, 25, 50 for DnCNN and ffdnet
}