Fix

open-mmlab · May 25, 2021 · ed9041c · ed9041c
1 parent 8f8027e
commit ed9041c
Showing 1 changed file with 18 additions and 3 deletions.
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
@@ -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',