Add and use rmsDiffNumpy

2024-03-25 20:14:33 +01:00
parent 7c7cbad0ef
commit 2baf9c3000
2 changed files with 7 additions and 3 deletions
--- a/algorithms/distance/rms_diff.py
+++ b/algorithms/distance/rms_diff.py
@@ -2,6 +2,7 @@ from PIL import ImageChops
 import math
 import operator
 import functools
+import numpy as np

 def rmsDiffPil(im1, im2):
    "Calculate the root-mean-square difference between two images"
@@ -13,3 +14,5 @@ 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):
+    return np.sqrt(np.mean(np.square(image0 - image1)))