File size: 1,430 Bytes
9279e0d |
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 |
# ------------------------------------------------------------------------
# Copyright (c) 2022 megvii-model. All Rights Reserved.
# ------------------------------------------------------------------------
# Modified from BasicSR (https://github.com/xinntao/BasicSR)
# Copyright 2018-2020 BasicSR Authors
# ------------------------------------------------------------------------
# general settings
name: Baseline-GoPro-width32-test
model_type: ImageRestorationModel
scale: 1
num_gpu: 1 # set num_gpu: 0 for cpu mode
manual_seed: 10
# dataset and data loader settings
datasets:
test:
name: gopro-test
type: PairedImageDataset
dataroot_gt: ./datasets/GoPro/test/target.lmdb
dataroot_lq: ./datasets/GoPro/test/input.lmdb
io_backend:
type: lmdb
# network structures
network_g:
type: BaselineLocal
width: 32
enc_blk_nums: [1, 1, 1, 28]
middle_blk_num: 1
dec_blk_nums: [1, 1, 1, 1]
dw_expand: 1
ffn_expand: 2
# path
path:
pretrain_network_g: experiments/pretrained_models/Baseline-GoPro-width32.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_ssim
crop_border: 0
test_y_channel: false
# dist training settings
dist_params:
backend: nccl
port: 29500
|