Add algorithms/distance/rmsdiff.py

2024-03-22 13:59:41 +01:00 · 2024-03-22 13:59:41 +01:00 · 72e37a252a
commit 72e37a252a
parent cf22ff2694
1 changed files with 14 additions and 0 deletions
--- a/algorithms/distance/rmsdiff.py
+++ b/algorithms/distance/rmsdiff.py
@ -0,0 +1,14 @@
+from PIL import ImageChops
+import math
+import operator
+import functools
+
+def rmsdiff(im1, im2):
+    "Calculate the root-mean-square difference between two images"
+
+    h = ImageChops.difference(im1, im2).histogram()
+
+    # calculate rms
+    return math.sqrt(functools.reduce(operator.add,
+        map(lambda h, i: h*(i**2), h, range(256))
+    ) / (float(im1.size[0]) * im1.size[1]))