File size: 2,120 Bytes
0a4be84 8cb8316 029f910 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 9c116e0 0a4be84 8cb8316 0a4be84 8cb8316 0a4be84 | 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 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | # general settings
name: test_DAT_2_x4
model_type: DATModel
scale: 4
num_gpu: 1
manual_seed: 10
datasets:
test_1: # the 1st test dataset
task: SR
name: Set5
type: PairedImageDataset
dataroot_gt: datasets/benchmark/Set5/HR
dataroot_lq: datasets/benchmark/Set5/LR_bicubic/X4
filename_tmpl: '{}x4'
io_backend:
type: disk
test_2: # the 2st test dataset
task: SR
name: Set14
type: PairedImageDataset
dataroot_gt: datasets/benchmark/Set14/HR
dataroot_lq: datasets/benchmark/Set14/LR_bicubic/X4
filename_tmpl: '{}x4'
io_backend:
type: disk
test_3: # the 3st test dataset
task: SR
name: B100
type: PairedImageDataset
dataroot_gt: datasets/benchmark/B100/HR
dataroot_lq: datasets/benchmark/B100/LR_bicubic/X4
filename_tmpl: '{}x4'
io_backend:
type: disk
test_4: # the 4st test dataset
task: SR
name: Urban100
type: PairedImageDataset
dataroot_gt: datasets/benchmark/Urban100/HR
dataroot_lq: datasets/benchmark/Urban100/LR_bicubic/X4
filename_tmpl: '{}x4'
io_backend:
type: disk
test_5: # the 5st test dataset
task: SR
name: Manga109
type: PairedImageDataset
dataroot_gt: datasets/benchmark/Manga109/HR
dataroot_lq: datasets/benchmark/Manga109/LR_bicubic/X4
filename_tmpl: '{}_LRBI_x4'
io_backend:
type: disk
# network structures
network_g:
type: DAT
upscale: 4
in_chans: 3
img_size: 64
img_range: 1.
split_size: [8,32]
depth: [6,6,6,6,6,6]
embed_dim: 180
num_heads: [6,6,6,6,6,6]
expansion_factor: 2
resi_connection: '1conv'
# path
path:
pretrain_network_g: experiments/pretrained_models/DAT-2/DAT_2_x4.pth
strict_load_g: True
# validation settings
val:
save_img: True
suffix: ~ # add suffix to saved images, if None, use exp name
use_chop: False # True to save memory, if img too large
metrics:
psnr: # metric name, can be arbitrary
type: calculate_psnr
crop_border: 4
test_y_channel: True
ssim:
type: calculate_ssim
crop_border: 4
test_y_channel: True
|