Fix RESISC45 dataset stats (#1349)

microsoft · May 19, 2023 · 2bf52eb · 2bf52eb
1 parent bbf6108
commit 2bf52eb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torchgeo/datamodules/resisc45.py b/torchgeo/datamodules/resisc45.py
@@ -19,8 +19,9 @@ class RESISC45DataModule(NonGeoDataModule):
     Uses the train/val/test splits from the dataset.
     """
 
-    mean = torch.tensor([127.86820969, 127.88083247, 127.84341029])
-    std = torch.tensor([51.8668062, 47.2380768, 47.0613924])
+    # Computed on the train set
+    mean = torch.tensor([93.89391792, 97.11226906, 87.56775284])
+    std = torch.tensor([51.84919672, 47.2365918, 47.06308786])
 
     def __init__(
         self, batch_size: int = 64, num_workers: int = 0, **kwargs: Any