Add back SSIM to configuration

open-mmlab · Apr 25, 2021 · 3c1612e · 3c1612e
1 parent 9d30c03
commit 3c1612e
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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'