diff --git a/options/train/GoPro/Baseline-width64.yml b/options/train/GoPro/Baseline-width64.yml new file mode 100644 index 0000000..ef82ad2 --- /dev/null +++ b/options/train/GoPro/Baseline-width64.yml @@ -0,0 +1,110 @@ +# ------------------------------------------------------------------------ +# 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-width64 +model_type: ImageRestorationModel +scale: 1 +num_gpu: 8 +manual_seed: 10 + +datasets: + train: + name: gopro-train + type: PairedImageDataset + dataroot_gt: ./datasets/GoPro/train/sharp_crops.lmdb + dataroot_lq: ./datasets/GoPro/train/blur_crops.lmdb + + filename_tmpl: '{}' + io_backend: + type: lmdb + + gt_size: 256 + use_flip: true + use_rot: true + + # data loader + use_shuffle: true + num_worker_per_gpu: 8 + batch_size_per_gpu: 8 + dataset_enlarge_ratio: 1 + prefetch_mode: ~ + + val: + name: gopro-test + type: PairedImageDataset + dataroot_gt: ./datasets/GoPro/test/target.lmdb + dataroot_lq: ./datasets/GoPro/test/input.lmdb + io_backend: + type: lmdb + + +network_g: + type: BaselineLocal + width: 64 + enc_blk_nums: [1, 1, 1, 28] + middle_blk_num: 1 + dec_blk_nums: [1, 1, 1, 1] + dw_expand: 2 + ffn_expand: 2 + +# path +path: + pretrain_network_g: ~ + strict_load_g: true + resume_state: ~ + +# training settings +train: + optim_g: + type: AdamW + lr: !!float 1e-3 + weight_decay: 0. + betas: [0.9, 0.9] + + scheduler: + type: TrueCosineAnnealingLR + T_max: 400000 + eta_min: !!float 1e-7 + + total_iter: 400000 + warmup_iter: -1 # no warm up + + # losses + pixel_opt: + type: PSNRLoss + loss_weight: 1 + reduction: mean + +# validation settings +val: + val_freq: !!float 2e4 + save_img: 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 + +# logging settings +logger: + print_freq: 200 + save_checkpoint_freq: !!float 5e3 + use_tb_logger: true + wandb: + project: ~ + resume_id: ~ + +# dist training settings +dist_params: + backend: nccl + port: 29500 diff --git a/options/train/SIDD/Baseline-width64.yml b/options/train/SIDD/Baseline-width64.yml new file mode 100644 index 0000000..fffc9a0 --- /dev/null +++ b/options/train/SIDD/Baseline-width64.yml @@ -0,0 +1,110 @@ +# ------------------------------------------------------------------------ +# 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-SIDD-width64 +model_type: ImageRestorationModel +scale: 1 +num_gpu: 8 +manual_seed: 10 + +datasets: + train: + name: SIDD + type: PairedImageDataset + dataroot_gt: ./datasets/SIDD/train/gt_crops.lmdb + dataroot_lq: ./datasets/SIDD/train/input_crops.lmdb + + filename_tmpl: '{}' + io_backend: + type: lmdb + + gt_size: 256 + use_flip: false + use_rot: false + + # data loader + use_shuffle: true + num_worker_per_gpu: 8 + batch_size_per_gpu: 8 + dataset_enlarge_ratio: 1 + prefetch_mode: ~ + + val: + name: SIDD_val + type: PairedImageDataset + dataroot_gt: ./datasets/SIDD/val/gt_crops.lmdb + dataroot_lq: ./datasets/SIDD/val/input_crops.lmdb + io_backend: + type: lmdb + + +network_g: + type: Baseline + width: 64 + enc_blk_nums: [2, 2, 4, 8] + middle_blk_num: 12 + dec_blk_nums: [2, 2, 2, 2] + dw_expand: 2 + ffn_expand: 2 + +# path +path: + pretrain_network_g: ~ + strict_load_g: true + resume_state: ~ + +# training settings +train: + optim_g: + type: AdamW + lr: !!float 1e-3 + weight_decay: 0. + betas: [0.9, 0.9] + + scheduler: + type: TrueCosineAnnealingLR + T_max: 400000 + eta_min: !!float 1e-7 + + total_iter: 400000 + warmup_iter: -1 # no warm up + + # losses + pixel_opt: + type: PSNRLoss + loss_weight: 1 + reduction: mean + +# validation settings +val: + val_freq: !!float 2e4 + save_img: false + use_image: 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 + +# logging settings +logger: + print_freq: 200 + save_checkpoint_freq: !!float 5e3 + use_tb_logger: true + wandb: + project: ~ + resume_id: ~ + +# dist training settings +dist_params: + backend: nccl + port: 29500