forked from mrq/DL-Art-School
28 lines
586 B
YAML
28 lines
586 B
YAML
|
name: RRDB_ESRGAN_x4
|
||
|
|
suffix: ~ # add suffix to saved images
|
||
|
|
model: sr
|
||
|
|
distortion: sr
|
||
|
|
scale: 4
|
||
|
|
crop_border: ~ # crop border when evaluation. If None(~), crop the scale pixels
|
||
|
|
gpu_ids: [0]
|
||
|
|
|
||
|
|
datasets:
|
||
|
|
test_1: # the 1st test dataset
|
||
|
|
name: set5
|
||
|
|
mode: LQ
|
||
|
|
dataroot_LQ: ../datasets/upsample_tests
|
||
|
|
|
||
|
|
#### network structures
|
||
|
|
network_G:
|
||
|
|
which_model_G: RRDBNet
|
||
|
|
in_nc: 3
|
||
|
|
out_nc: 3
|
||
|
|
nf: 64
|
||
|
|
nb: 23
|
||
|
|
upscale: 4
|
||
|
|
|
||
|
|
#### path
|
||
|
|
path:
|
||
|
|
pretrain_model_G: ../experiments/ESRGANx4_blacked_ft/models/31500_G.pth
|
||
|
|
pretrain_model_D: ../experiments/ESRGANx4_blacked_ft/models/31500_D.pth
|