diff --git a/configs/restorers/basicvsr/basicvsr_reds4.py b/configs/restorers/basicvsr/basicvsr_reds4.py index e90903bd59..5a05ee33f0 100644 --- a/configs/restorers/basicvsr/basicvsr_reds4.py +++ b/configs/restorers/basicvsr/basicvsr_reds4.py @@ -12,7 +12,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0) +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0) # dataset settings train_dataset_type = 'SRREDSMultipleGTDataset' diff --git a/configs/restorers/basicvsr/basicvsr_vimeo90k_bd.py b/configs/restorers/basicvsr/basicvsr_vimeo90k_bd.py index 3ca2cb4931..f09980b512 100644 --- a/configs/restorers/basicvsr/basicvsr_vimeo90k_bd.py +++ b/configs/restorers/basicvsr/basicvsr_vimeo90k_bd.py @@ -12,7 +12,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0, convert_to='y') +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0, convert_to='y') # dataset settings train_dataset_type = 'SRVimeo90KMultipleGTDataset' diff --git a/configs/restorers/basicvsr/basicvsr_vimeo90k_bi.py b/configs/restorers/basicvsr/basicvsr_vimeo90k_bi.py index 7b0be4f181..e58369eb42 100644 --- a/configs/restorers/basicvsr/basicvsr_vimeo90k_bi.py +++ b/configs/restorers/basicvsr/basicvsr_vimeo90k_bi.py @@ -12,7 +12,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0, convert_to='y') +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0, convert_to='y') # dataset settings train_dataset_type = 'SRVimeo90KMultipleGTDataset' diff --git a/configs/restorers/iconvsr/iconvsr_reds4.py b/configs/restorers/iconvsr/iconvsr_reds4.py index 43e7d17b06..e99336e24c 100644 --- a/configs/restorers/iconvsr/iconvsr_reds4.py +++ b/configs/restorers/iconvsr/iconvsr_reds4.py @@ -16,7 +16,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0) +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0) # dataset settings train_dataset_type = 'SRREDSMultipleGTDataset' diff --git a/configs/restorers/iconvsr/iconvsr_vimeo90k_bd.py b/configs/restorers/iconvsr/iconvsr_vimeo90k_bd.py index a657dee901..e0f41d1c79 100644 --- a/configs/restorers/iconvsr/iconvsr_vimeo90k_bd.py +++ b/configs/restorers/iconvsr/iconvsr_vimeo90k_bd.py @@ -16,7 +16,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0, convert_to='y') +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0, convert_to='y') # dataset settings train_dataset_type = 'SRVimeo90KMultipleGTDataset' diff --git a/configs/restorers/iconvsr/iconvsr_vimeo90k_bi.py b/configs/restorers/iconvsr/iconvsr_vimeo90k_bi.py index 9692d938b9..738a14b2ba 100644 --- a/configs/restorers/iconvsr/iconvsr_vimeo90k_bi.py +++ b/configs/restorers/iconvsr/iconvsr_vimeo90k_bi.py @@ -16,7 +16,7 @@ pixel_loss=dict(type='CharbonnierLoss', loss_weight=1.0, reduction='mean')) # model training and testing settings train_cfg = dict(fix_iter=5000) -test_cfg = dict(metrics=['PSNR'], crop_border=0, convert_to='y') +test_cfg = dict(metrics=['PSNR', 'SSIM'], crop_border=0, convert_to='y') # dataset settings train_dataset_type = 'SRVimeo90KMultipleGTDataset'