Peak Signal to Noise Ratio (PSNR) in Python for an Image

Question

How do I calculate the maximum signal to noise ratio (PSNR) in Python? Is there any library that can calculate PSNR for Image?

Answer 1

The steps for calculation of PSNR value of two images:

For details click here

import numpy 
import math
import cv2
original = cv2.imread("original.png")
contrast = cv2.imread("photoshopped.png",1)
def psnr(img1, img2):
    mse = numpy.mean( (img1 - img2) ** 2 )
    if mse == 0:
    return 100
    PIXEL_MAX = 255.0
    return 20 * math.log10(PIXEL_MAX / math.sqrt(mse))

d=psnr(original,contrast)
print(d)

Answer 2

You can use cv2.PSNR like this example:

import cv2
img1 = cv2.imread('img1.bmp')
img2 = cv2.imread('img2.bmp')
psnr = cv2.PSNR(img1, img2)

Answer 3

turn to float first!!!!!!!! turn to float first!!!!!!!! turn to float first!!!!!!!!

def compute_psnr(img1, img2):
img1 = img1.astype(np.float64) / 255.
img2 = img2.astype(np.float64) / 255.
mse = np.mean((img1 - img2) ** 2)
if mse == 0:
    return "Same Image"
return 10 * math.log10(1. / mse)

Answer 4

If you have a replica of your signal (image) that is noise free, you can calculate the correlation coefficient which is directly related to SNR.

See my response here for specific details on determining the correlation coefficient and from that SNR:

Noise detection

In this context there is no "maximum SNR" but will be the SNR for your entire image, meaning the power of your desired signal relative to everything else (distortions).