Python OpenCV 平均二乗誤差の算出

Question

mse = (np.square(img1 - img2)).mean(axis=None)

https://www.366service.com/jp/qa/3e40bbe248e62f6212ad603a0b911c98

2画像のMSEを算出したいと思い、調べたところ上記のもので算出できるとのことなのですが
自分で算出のためのコードを書いて実行したところ上記のコードの算出結果と異なる値となりました。
以下がそのコードです。処理の違いを教えてください。

import cv2
import numpy as np

img1 = cv2.imread("img1",0)
img2 = cv2.imread("img1",0)

sum1 = 0
l1=list(range(img1.shape[0]))
l2=list(range(img1.shape[1]))

for i in l1:
    for j in l2:
        m1 = int(img1[i][j])
        m2 = int(img2[i][j])
        m = (m1-m2) ** 2
        sum1 += m

MSE = sum1/(img1.shape[0]*img1.shape[1])
print(MSE)

Answer 1

img1, img2 は uint8 型で 0 ~ 255 の値しか表現できないので、減算結果が負の値になったときに計算結果がおかしくなっています。
以下のように修正すれば同じ値になります。

- mse = (np.square(img1 - img2)).mean(axis=None)
+ mse = (np.square(img1.astype(int) - img2.astype(int))).mean(axis=None)

Comments

[nakamura1]

解決しました！ありがとうございます！！