deep-tempest/end-to-end/options/train_drunet_denoiser.json

{
  "task": "drunet"  //  root/task/images-models-options
  , "model": "plain" // "plain"
  , "gpu_ids": [0]

  , "scale": 0       // broadcast to "netG" if SISR
  , "n_channels": 1  // broadcast to "datasets", 1 for grayscale, 3 for color
  , "n_channels_datasetload": 3 // broadcast to image training set
  , "sigma": [0, 50]      // 15, 25, 50 for DnCNN | [0, 75] for FFDNet and FDnCNN
  , "sigma_test": 5 // 15, 25, 50 for DnCNN and ffdnet

  , "path": {
    "root": "denoising"            // "denoising" | "superresolution"
    , "pretrained_netG": null    // 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": "trainsets/web_images_train"// path of H training dataset
      , "num_patches_per_image": 20   // number of random patches of training image
      , "dataroot_L": "trainsets/simulations" // path of L training dataset, if using noisy H type: null
      , "H_size": 128                    // patch size 40 | 64 | 96 | 128 | 192
      , "dataloader_shuffle": true
      , "dataloader_num_workers": 8
      , "dataloader_batch_size": 64     // batch size 1 | 16 | 32 | 48 | 64 | 128
    }
    , "test": {
      "name": "test_dataset"            // just name
      , "dataset_type": "ffdnet"         // "dncnn" | "dnpatch" for dncnn,  | "fdncnn" | "ffdnet" | "sr" | "srmd" | "dpsr" | "plain" | "plainpatch"
      , "dataroot_H": "testsets/web_images_test"  // path of H testing dataset
      , "dataroot_L": "testsets/simulations"   // path of L testing dataset
    }
  }

  , "netG": {
    "net_type": "drunet" // "dncnn" | "fdncnn" | "ffdnet" | "srmd" | "dpsr" | "srresnet0" |  "srresnet1" | "rrdbnet" 
    , "in_nc": 1        // input channel number
    , "out_nc": 1       // ouput channel number
    , "nc": [64, 128, 256, 512]          // 64 for "dncnn"
    , "nb": 4          // 17 for "dncnn", 20 for dncnn3, 16 for "srresnet"
    , "gc": 32          // unused
    , "ng": 2           // unused
    , "reduction": 16  // unused
    , "act_mode": "R"  // "BR" for BN+ReLU | "R" for ReLU
    , "upsample_mode": "convtranspose"  // "pixelshuffle" | "convtranspose" | "upconv"
    , "downsample_mode": "strideconv"   // "strideconv" | "avgpool" | "maxpool"
    , "bias": false//
    , "init_type": "orthogonal"         // "orthogonal" | "normal" | "uniform" | "xavier_normal" | "xavier_uniform" | "kaiming_normal" | "kaiming_uniform"
    , "init_bn_type": "uniform"         // "uniform" | "constant"
    , "init_gain": 0.2
  }

  , "train": {
      "epochs": 301                      // number of epochs to train
    , "G_lossfn_type": "l1"               // "l1" preferred | "l2sum" | "l2" | "ssim" 
    , "G_lossfn_weight": 1.0            // default
    , "G_tvloss_weight": 0.1            // total variation weight

    , "G_optimizer_type": "adam"        // fixed, adam is enough
    , "G_optimizer_lr": 1e-4           // learning rate
    , "G_optimizer_clipgrad": null      // unused

    , "G_scheduler_type": "MultiStepLR" // "MultiStepLR" is enough
    , "G_scheduler_milestones": [640, 980, 1600, 1920, 2400, 4800, 6400, 9280]
    , "G_scheduler_gamma": 0.1  //

    , "G_regularizer_orthstep": null    // unused
    , "G_regularizer_clipstep": null    // unused
    
    // iteration (batch step) checkpoints
    , "checkpoint_test": 320           // for testing
    , "checkpoint_save": 1600           // for saving model
    , "checkpoint_print": 32           // for print
  }
}