86 lines
2.1 KiB
YAML
86 lines
2.1 KiB
YAML
# ------------------------------------------------------------------------
|
|
# 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_4x
|
|
model_type: ImageRestorationModel
|
|
scale: 4
|
|
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_x4
|
|
io_backend:
|
|
type: disk
|
|
|
|
test1:
|
|
name: KITTI2015
|
|
type: PairedStereoImageDataset
|
|
dataroot_gt: datasets/StereoSR/test/KITTI2015/hr
|
|
dataroot_lq: datasets/StereoSR/test/KITTI2015/lr_x4
|
|
io_backend:
|
|
type: disk
|
|
|
|
test2:
|
|
name: Middlebury
|
|
type: PairedStereoImageDataset
|
|
dataroot_gt: datasets/StereoSR/test/Middlebury/hr
|
|
dataroot_lq: datasets/StereoSR/test/Middlebury/lr_x4
|
|
io_backend:
|
|
type: disk
|
|
|
|
test3:
|
|
name: Flickr1024
|
|
type: PairedStereoImageDataset
|
|
dataroot_gt: datasets/StereoSR/test/Flickr1024/hr
|
|
dataroot_lq: datasets/StereoSR/test/Flickr1024/lr_x4
|
|
io_backend:
|
|
type: disk
|
|
|
|
# network structures
|
|
network_g:
|
|
type: NAFSSR
|
|
up_scale: 4
|
|
width: 128
|
|
num_blks: 128
|
|
|
|
|
|
# path
|
|
path:
|
|
pretrain_network_g: experiments/pretrained_models/NAFSSR-L_4x.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
|