Tiny Fix

open-mmlab · Apr 27, 2021 · 8f0c8f8 · 8f0c8f8
1 parent 0cba98c
commit 8f0c8f8
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 48 deletions.
diff --git a/configs/restorers/rdn/rdn_x2c64b16_g1_1000k_div2k.py b/configs/restorers/rdn/rdn_x2c64b16_g1_1000k_div2k.py
@@ -24,19 +24,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='PairedRandomCrop', gt_patch_size=64),
     dict(
         type='Flip', keys=['lq', 'gt'], flip_ratio=0.5,
@@ -51,19 +47,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='Collect', keys=['lq', 'gt'], meta_keys=['lq_path', 'gt_path']),
     dict(type='ImageToTensor', keys=['lq', 'gt'])
 ]

diff --git a/configs/restorers/rdn/rdn_x3c64b16_g1_1000k_div2k.py b/configs/restorers/rdn/rdn_x3c64b16_g1_1000k_div2k.py
@@ -24,19 +24,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='PairedRandomCrop', gt_patch_size=96),
     dict(
         type='Flip', keys=['lq', 'gt'], flip_ratio=0.5,
@@ -51,19 +47,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='Collect', keys=['lq', 'gt'], meta_keys=['lq_path', 'gt_path']),
     dict(type='ImageToTensor', keys=['lq', 'gt'])
 ]

diff --git a/configs/restorers/rdn/rdn_x4c64b16_g1_1000k_div2k.py b/configs/restorers/rdn/rdn_x4c64b16_g1_1000k_div2k.py
@@ -24,19 +24,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='PairedRandomCrop', gt_patch_size=128),
     dict(
         type='Flip', keys=['lq', 'gt'], flip_ratio=0.5,
@@ -51,19 +47,15 @@
         type='LoadImageFromFile',
         io_backend='disk',
         key='lq',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(
         type='LoadImageFromFile',
         io_backend='disk',
         key='gt',
-        flag='unchanged'),
+        flag='color',
+        channel_order='rgb'),
     dict(type='RescaleToZeroOne', keys=['lq', 'gt']),
-    dict(
-        type='Normalize',
-        keys=['lq', 'gt'],
-        mean=[0, 0, 0],
-        std=[1, 1, 1],
-        to_rgb=True),
     dict(type='Collect', keys=['lq', 'gt'], meta_keys=['lq_path', 'gt_path']),
     dict(type='ImageToTensor', keys=['lq', 'gt'])
 ]