Add and use getColorMeans

datasets/raise/utils.py

@@ -7,6 +7,7 @@ from PIL import Image
 from skimage import img_as_float
 from datetime import datetime
 import builtins as __builtin__
+from scipy.ndimage import gaussian_filter
 
 class Color(Enum):
     RED = auto()
@@ -130,3 +131,14 @@ def updateExtremes(imageNpArray, minColor, maxColor):
 
 def print(*toPrint):
     __builtin__.print(datetime.now(), *toPrint)
+
+def getColorMeans(imagesFileNames, colors = Color):
+    colorMeans = {}
+    for color in colors:
+        colorIterativeMean = iterativeMean()
+        for imageFileName in tqdm(imagesFileNames, f'Computing mean of {color} colored images'):
+            imageNpArray = getImageNpArray(imageFileName, False, color)
+            imageNpArray = gaussian_filter(imageNpArray, sigma = 5)
+            colorIterativeMean.add(imageNpArray)
+        colorMeans[color] = colorIterativeMean.mean
+    return colorMeans