| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115 |
- scale: 4
- num_gpu: 1
- manual_seed: 0
- is_train: True
- dist: False
- # ----------------- options for synthesizing training data ----------------- #
- # USM the ground-truth
- l1_gt_usm: True
- percep_gt_usm: True
- gan_gt_usm: False
- # the first degradation process
- resize_prob: [0.2, 0.7, 0.1] # up, down, keep
- resize_range: [0.15, 1.5]
- gaussian_noise_prob: 1
- noise_range: [1, 30]
- poisson_scale_range: [0.05, 3]
- gray_noise_prob: 1
- jpeg_range: [30, 95]
- # the second degradation process
- second_blur_prob: 1
- resize_prob2: [0.3, 0.4, 0.3] # up, down, keep
- resize_range2: [0.3, 1.2]
- gaussian_noise_prob2: 1
- noise_range2: [1, 25]
- poisson_scale_range2: [0.05, 2.5]
- gray_noise_prob2: 1
- jpeg_range2: [30, 95]
- gt_size: 32
- queue_size: 1
- # network structures
- network_g:
- type: RRDBNet
- num_in_ch: 3
- num_out_ch: 3
- num_feat: 4
- num_block: 1
- num_grow_ch: 2
- network_d:
- type: UNetDiscriminatorSN
- num_in_ch: 3
- num_feat: 2
- skip_connection: True
- # path
- path:
- pretrain_network_g: ~
- param_key_g: params_ema
- strict_load_g: true
- resume_state: ~
- # training settings
- train:
- ema_decay: 0.999
- optim_g:
- type: Adam
- lr: !!float 1e-4
- weight_decay: 0
- betas: [0.9, 0.99]
- optim_d:
- type: Adam
- lr: !!float 1e-4
- weight_decay: 0
- betas: [0.9, 0.99]
- scheduler:
- type: MultiStepLR
- milestones: [400000]
- gamma: 0.5
- total_iter: 400000
- warmup_iter: -1 # no warm up
- # losses
- pixel_opt:
- type: L1Loss
- loss_weight: 1.0
- reduction: mean
- # perceptual loss (content and style losses)
- perceptual_opt:
- type: PerceptualLoss
- layer_weights:
- # before relu
- 'conv1_2': 0.1
- 'conv2_2': 0.1
- 'conv3_4': 1
- 'conv4_4': 1
- 'conv5_4': 1
- vgg_type: vgg19
- use_input_norm: true
- perceptual_weight: !!float 1.0
- style_weight: 0
- range_norm: false
- criterion: l1
- # gan loss
- gan_opt:
- type: GANLoss
- gan_type: vanilla
- real_label_val: 1.0
- fake_label_val: 0.0
- loss_weight: !!float 1e-1
- net_d_iters: 1
- net_d_init_iters: 0
- # validation settings
- val:
- val_freq: !!float 5e3
- save_img: False
|
