From 8f0c8f8e64a56a73e61654a23fd604fe9c1888d7 Mon Sep 17 00:00:00 2001
From: liyinshuo <liyinshuo@sensetime.com>
Date: Tue, 27 Apr 2021 09:58:19 +0800
Subject: [PATCH] Tiny Fix

---
 .../rdn/rdn_x2c64b16_g1_1000k_div2k.py        | 24 +++++++------------
 .../rdn/rdn_x3c64b16_g1_1000k_div2k.py        | 24 +++++++------------
 .../rdn/rdn_x4c64b16_g1_1000k_div2k.py        | 24 +++++++------------
 3 files changed, 24 insertions(+), 48 deletions(-)

diff --git a/configs/restorers/rdn/rdn_x2c64b16_g1_1000k_div2k.py b/configs/restorers/rdn/rdn_x2c64b16_g1_1000k_div2k.py
index ee7187fe34..7aab0fc7a3 100644
--- 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
index a609cba2fd..fc8784d339 100644
--- 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
index 78d5677d1d..390e2bb5cb 100644
--- 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'])
 ]