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)

'AI > CV' 카테고리의 다른 글

[이미지 품질 평가] SSIM  (0) 2022.06.30