Spaces:
Running
Running
File size: 2,172 Bytes
2514fb4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
{
"task": "usrnet" //
, "model": "plain4" // "plain" | "gan"
, "gpu_ids": [0]
, "scale": 4
, "n_channels": 3 // 1 for grayscale image restoration, 3 for color image restoration
, "merge_bn": false
, "merge_bn_startpoint": 300000
, "datasets": {
"train": {
"name": "train_dataset"
, "dataset_type": "usrnet"
, "dataroot_H": "trainsets/trainH"
, "dataroot_L": null
, "H_size": 96 // 128 | 192
, "use_flip": true
, "use_rot": true
, "scales": [1, 2, 3, 4]
, "dataloader_shuffle": true
, "dataloader_num_workers": 8
, "dataloader_batch_size": 48
}
, "test": {
"name": "test_dataset"
, "dataset_type": "usrnet"
, "dataroot_H": "testsets/set5"
, "dataroot_L": null
}
}
, "path": {
"root": "SR"
, "pretrained_netG": null
}
, "netG": {
"net_type": "usrnet" // "srresnet" | "rrdbnet" | "rcan" | "unet" | "unetplus" | "nonlocalunet"
, "n_iter": 6 // 8
, "h_nc": 32 // 64
, "in_nc": 4
, "out_nc": 3
, "nc": [16, 32, 64, 64] // [64, 128, 256, 512] for "unet"
, "nb": 2
, "gc": 32
, "ng": 2
, "reduction" : 16
, "act_mode": "R" // "BR" for BN+ReLU | "R" for ReLU
, "upsample_mode": "convtranspose" // "pixelshuffle" | "convtranspose" | "upconv"
, "downsample_mode": "strideconv" // "strideconv" | "avgpool" | "maxpool"
, "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": {
"G_lossfn_type": "l1" // "l1" | "l2sum" | "l2" | "ssim"
, "G_lossfn_weight": 1.0
, "G_optimizer_type": "adam"
, "G_optimizer_lr": 1e-4
, "G_optimizer_wd": 0
, "G_optimizer_clipgrad": null
, "G_scheduler_type": "MultiStepLR"
, "G_scheduler_milestones": [100000, 200000, 300000, 400000]
, "G_scheduler_gamma": 0.5
, "G_regularizer_orthstep": null
, "G_regularizer_clipstep": null
, "checkpoint_test": 5000
, "checkpoint_save": 5000
, "checkpoint_print": 200
}
}
|