Fix crash when images are identical

mrebersv · Mar 11, 2015 · 71d6c9c · 71d6c9c
1 parent 41ec2df
commit 71d6c9c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/psnr.py b/psnr.py
@@ -20,6 +20,8 @@
 import math
 
 def psnr(img1, img2):
-	mse = numpy.mean( (img1 - img2) ** 2 )
-	PIXEL_MAX = 255.0
-	return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))
+    mse = numpy.mean( (img1 - img2) ** 2 )
+    if mse == 0:
+        return 100
+    PIXEL_MAX = 255.0
+    return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))