PSNRについて教えてください

Question

HDR画像の評価にPSNRを使いたいのですが、プログラムがわかりません。

具体的に行いたいことは「.hdr」と「.exr」を比べて評価することです。

その際にこちらのサイト[python]画質評価としてのPSNRのプログラムを使いました。プログラムは下です。
サイトでは8bitだったのでPSNRを求めるところが255だったので、16bitの65535に書き直しました。

import cv2
import numpy as np
import math

#画像の読み込み
Original  = cv2.imread('Test1.hdr') #元画像
Distorted = cv2.imread('Redu1.exr') #再構築した画像

#画素値の読み込み
pixel_value_Ori = Original.flatten().astype(float)
pixel_value_Dis = Distorted.flatten().astype(float)

#画素情報の取得
imageHeight, imageWidth, BPP = Original.shape

#画素数
N = imageHeight * imageWidth 

#1画素あたりRGB3つの情報がある. 
addr = N * BPP

#RGB画素値の差の2乗の総和
sumR=0
sumG=0
sumB=0

#差の2乗の総和を計算
for i in range(addr):
    if(i%3==0):
        sumB += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 ) 
    elif(i%3==1):
        sumG += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 ) 
    else:
        sumR += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 ) 

#PSNRを求める
MSE =(sumR + sumG + sumB) / (3 * N )
PSNR = 10 * math.log(65535*65535/MSE,10)
print('PSNR',PSNR)

一応このプログラム自体は動きました。

ここでわからないのは、この数値であっているのかと言うことです。
「.hdr」と「.exr」と拡張子が違うためプログラム中の数値を変えただけでいいのでしょうか？
具体的にどうすればいいのか教えていただきたいです。
何卒お願い致します。

Comments

[taront]

この辺が原因かもしれませんね。

https://qiita.com/studio_haneya/items/3adeae5fe527023eed11

両方とも事前にjpgなどに変換してから実施してはどうでしょうか。

Answer 1

「.hdr」と「.exr」は拡張子が違うから、そもそもその２つを比較すること自体正しくない。

実際に実行したところ、明らかにおかしい結果となった。
具体的には、ノイズが多い画像の方が良い値になった。

.exrと.exrを比べたところ、納得する結果が得られた。

「解決策はない」というのが答えでした。