Remove ImageNet1k mean values and organize losses into separate files

muslll · Apr 7, 2024 · 417432c · 417432c
1 parent f23e1ca
commit 417432c
Show file tree

Hide file tree

Showing 18 changed files with 490 additions and 455 deletions.
diff --git a/neosr/archs/atd_arch.py b/neosr/archs/atd_arch.py
@@ -927,7 +927,7 @@ def __init__(
             self.no_norm = None
 
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)

diff --git a/neosr/archs/craft_arch.py b/neosr/archs/craft_arch.py
@@ -661,7 +661,7 @@ def __init__(self,
         self.num_feat = num_feat
         self.num_out_ch = num_out_ch
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)

diff --git a/neosr/archs/dat_arch.py b/neosr/archs/dat_arch.py
@@ -751,7 +751,7 @@ def __init__(self,
         num_feat = 64
         self.img_range = img_range
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)

diff --git a/neosr/archs/hat_arch.py b/neosr/archs/hat_arch.py
@@ -740,7 +740,7 @@ def __init__(self,
         num_feat = 64
         self.img_range = img_range
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)

diff --git a/neosr/archs/hdsrnet_arch.py b/neosr/archs/hdsrnet_arch.py
@@ -335,7 +335,7 @@ class MeanShift(nn.Conv2d):
     def __init__(
         self,
         rgb_range,
-        rgb_mean=(0.4488, 0.4371, 0.4040),
+        rgb_mean=(0.5, 0.5, 0.5),
         rgb_std=(1.0, 1.0, 1.0),
         sign=-1,
     ):

diff --git a/neosr/archs/rgt_arch.py b/neosr/archs/rgt_arch.py
@@ -842,7 +842,7 @@ def __init__(
         num_feat = 64
         self.img_range = img_range
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)

diff --git a/neosr/archs/span_arch.py b/neosr/archs/span_arch.py
@@ -213,7 +213,7 @@ def __init__(self,
                  bias=True,
                  norm=False,
                  img_range=1.0,
-                 rgb_mean=(0.4488, 0.4371, 0.4040)
+                 rgb_mean=(0.5, 0.5, 0.5)
                  ):
         super(span, self).__init__()
 

diff --git a/neosr/archs/swinir_arch.py b/neosr/archs/swinir_arch.py
@@ -774,7 +774,7 @@ def __init__(self,
         num_feat = 64
         self.img_range = img_range
         if in_chans == 3:
-            rgb_mean = (0.4488, 0.4371, 0.4040)
+            rgb_mean = (0.5, 0.5, 0.5)
             self.mean = torch.Tensor(rgb_mean).view(1, 3, 1, 1)
         else:
             self.mean = torch.zeros(1, 1, 1, 1)