/
NAFSSR-L_2x.yml
85 lines (74 loc) · 2.07 KB
/
NAFSSR-L_2x.yml
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
# ------------------------------------------------------------------------
# Copyright (c) 2022 megvii-model. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from BasicSR (https://github.com/xinntao/BasicSR)
# Copyright 2018-2020 BasicSR Authors
# ------------------------------------------------------------------------
# general settings
name: NAFSSR-L_2x
model_type: ImageRestorationModel
scale: 2
num_gpu: 1 # set num_gpu: 0 for cpu mode
manual_seed: 10
# dataset and data loader settings
datasets:
test0:
name: KITTI2012
type: PairedStereoImageDataset
dataroot_gt: datasets/StereoSR/test/KITTI2012/hr
dataroot_lq: datasets/StereoSR/test/KITTI2012/lr_x2
io_backend:
type: disk
test1:
name: KITTI2015
type: PairedStereoImageDataset
dataroot_gt: datasets/StereoSR/test/KITTI2015/hr
dataroot_lq: datasets/StereoSR/test/KITTI2015/lr_x2
io_backend:
type: disk
test2:
name: Middlebury
type: PairedStereoImageDataset
dataroot_gt: datasets/StereoSR/test/Middlebury/hr
dataroot_lq: datasets/StereoSR/test/Middlebury/lr_x2
io_backend:
type: disk
test3:
name: Flickr1024
type: PairedStereoImageDataset
dataroot_gt: datasets/StereoSR/test/Flickr1024/hr
dataroot_lq: datasets/StereoSR/test/Flickr1024/lr_x2
io_backend:
type: disk
# network structures
network_g:
type: NAFSSR
up_scale: 2
width: 128
num_blks: 128
# path
path:
pretrain_network_g: experiments/pretrained_models/NAFSSR-L_2x.pth
strict_load_g: true
resume_state: ~
# validation settings
val:
save_img: true
grids: false
metrics:
psnr: # metric name, can be arbitrary
type: calculate_psnr
crop_border: 0
test_y_channel: false
ssim:
type: calculate_skimage_ssim
# psnr_left: # metric name, can be arbitrary
# type: calculate_psnr_left
# crop_border: 0
# test_y_channel: false
# ssim_left:
# type: calculate_skimage_ssim_left
# dist training settings
dist_params:
backend: nccl
port: 29500