64 lines
1.7 KiB
YAML
64 lines
1.7 KiB
YAML
dataset:
|
|
target: dataset.realesrgan.RealESRGANDataset
|
|
params:
|
|
# Path to the file list.
|
|
file_list: preset/train_datasets/train.list
|
|
out_size: 512
|
|
crop_type: center
|
|
|
|
use_hflip: true
|
|
use_rot: false
|
|
|
|
blur_kernel_size: 21
|
|
kernel_list: ['iso', 'aniso', 'generalized_iso', 'generalized_aniso', 'plateau_iso', 'plateau_aniso']
|
|
kernel_prob: [0.45, 0.25, 0.12, 0.03, 0.12, 0.03]
|
|
sinc_prob: 0.1
|
|
blur_sigma: [0.2, 1.5]
|
|
betag_range: [0.5, 2.0]
|
|
betap_range: [1, 1.5]
|
|
|
|
blur_kernel_size2: 11
|
|
kernel_list2: ['iso', 'aniso', 'generalized_iso', 'generalized_aniso', 'plateau_iso', 'plateau_aniso']
|
|
kernel_prob2: [0.45, 0.25, 0.12, 0.03, 0.12, 0.03]
|
|
sinc_prob2: 0.1
|
|
blur_sigma2: [0.2, 1.0]
|
|
betag_range2: [0.5, 2.0]
|
|
betap_range2: [1, 1.5]
|
|
|
|
final_sinc_prob: 0.8
|
|
|
|
data_loader:
|
|
batch_size: 16
|
|
shuffle: true
|
|
num_workers: 16
|
|
prefetch_factor: 2
|
|
drop_last: true
|
|
|
|
batch_transform:
|
|
target: dataset.batch_transform.RealESRGANBatchTransform
|
|
params:
|
|
use_sharpener: false
|
|
resize_hq: false
|
|
# Queue size of training pool, this should be multiples of batch_size.
|
|
queue_size: 192
|
|
|
|
# the first degradation process
|
|
resize_prob: [0.2, 0.7, 0.1] # up, down, keep
|
|
resize_range: [0.3, 1.5]
|
|
gaussian_noise_prob: 0.5
|
|
noise_range: [1, 15]
|
|
poisson_scale_range: [0.05, 2.0]
|
|
gray_noise_prob: 0.4
|
|
jpeg_range: [60, 95]
|
|
|
|
# the second degradation process
|
|
stage2_scale: 4
|
|
second_blur_prob: 0.5
|
|
resize_prob2: [0.3, 0.4, 0.3] # up, down, keep
|
|
resize_range2: [0.6, 1.2]
|
|
gaussian_noise_prob2: 0.5
|
|
noise_range2: [1, 12]
|
|
poisson_scale_range2: [0.05, 1.0]
|
|
gray_noise_prob2: 0.4
|
|
jpeg_range2: [60, 100]
|