AI/CV
[이미지 품질 평가] PSNR
리네엔
2022. 6. 30. 19:42
Peak Signal to Noise Ratio : 최대 신호 대 잡음 비
신호가 가질 수 있는 최대 전력에 대한 잡음의 전력을 나타낸 것.
이미지의 화질 손실 정보를 평가할 때 사용
MSE를 최적화하면 자연스럽게 높일 수 있는 평가지표.
만일 평가할 이미지가 서로 같다면 MSE가 0이되어 inf가 됨
$$PSNR = 10\log_{10}{MAX^{2} \over MSE}$$
python code
import numpy as np
import math
# input : numpy 배열
def PSNR(img1,img2) :
MSE = np.mean((img1-img2)**2)
if MSE == 0 :
return float('inf')
return 10*math.log10(255**2/MSE)