BUG: Fix incorrect pixel standard deviation calculation

bioscan-ml · Dec 7, 2024 · b24090a · b24090a
1 parent 53d82ad
commit b24090a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/bioscan_dataset/bioscan1m.py b/bioscan_dataset/bioscan1m.py
@@ -17,7 +17,7 @@
 from torchvision.datasets.vision import VisionDataset
 
 RGB_MEAN = torch.tensor([0.72510918, 0.72891550, 0.72956181])
-RGB_STDEV = torch.tensor([0.66364000, 0.66088159, 0.66035860])
+RGB_STDEV = torch.tensor([0.12654378, 0.14301962, 0.16103319])
 
 COLUMN_DTYPES = {
     "sampleid": str,

diff --git a/bioscan_dataset/bioscan5m.py b/bioscan_dataset/bioscan5m.py
@@ -18,7 +18,8 @@
 from torchvision.datasets.vision import VisionDataset
 
 RGB_MEAN = torch.tensor([0.76281859, 0.76503749, 0.76373138])
-RGB_STDEV = torch.tensor([0.67122520, 0.66740192, 0.66488950])
+RGB_STDEV = torch.tensor([0.14205404, 0.15642502, 0.17309470])
+
 
 COLUMN_DTYPES = {
     "processid": str,