Show both RMS and RMS normalized

2024-03-29 01:47:01 +01:00
parent 3ccec5bbd0
commit 2f66e82f22
2 changed files with 11 additions and 3 deletions
--- a/algorithms/distance/rms_diff.py
+++ b/algorithms/distance/rms_diff.py
@@ -14,5 +14,13 @@ def rmsDiffPil(im1, im2):
        map(lambda h, i: h*(i**2), h, range(256))
    ) / (float(im1.size[0]) * im1.size[1]))

-def rmsDiffNumpy(image0, image1):
+def normalizeImage(image):
+    image = image - image.min()
+    image = image / image.max()
+    return image
+
+def rmsDiffNumpy(image0, image1, normalize = False):
+    if normalize:
+        image0 = normalizeImage(image0)
+        image1 = normalizeImage(image1)
    return np.sqrt(np.mean(np.square(image0 - image1)))