diff --git a/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py b/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py index 786f62d3f9..9a1ac0ef11 100644 --- a/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py +++ b/configs/restorers/ttsr/ttsr-rec_x4_c64b16_g1_200k_CUFED.py @@ -55,9 +55,14 @@ backend='pillow'), dict( type='Normalize', - keys=['lq', 'gt', 'lq_up', 'ref', 'ref_downup'], + keys=['lq', 'gt'], mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5]), + dict( + type='Normalize', + keys=['lq_up', 'ref', 'ref_downup'], + mean=[0., 0., 0.], + std=[255., 255., 255.]), dict( type='Flip', keys=['lq', 'gt', 'lq_up'], @@ -127,9 +132,14 @@ backend='pillow'), dict( type='Normalize', - keys=['lq', 'gt', 'lq_up', 'ref', 'ref_downup'], + keys=['lq', 'gt'], mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5]), + dict( + type='Normalize', + keys=['lq_up', 'ref', 'ref_downup'], + mean=[0., 0., 0.], + std=[255., 255., 255.]), dict( type='ImageToTensor', keys=['lq', 'gt', 'lq_up', 'ref', 'ref_downup']), dict( @@ -170,9 +180,14 @@ backend='pillow'), dict( type='Normalize', - keys=['lq', 'lq_up', 'ref', 'ref_downup'], + keys=['lq'], mean=[127.5, 127.5, 127.5], std=[127.5, 127.5, 127.5]), + dict( + type='Normalize', + keys=['lq_up', 'ref', 'ref_downup'], + mean=[0., 0., 0.], + std=[255., 255., 255.]), dict(type='ImageToTensor', keys=['lq', 'lq_up', 'ref', 'ref_downup']), dict( type='Collect',