IndexError: index 1658880 is out of bounds for axis 0 with size 1658880

Question

[python]画質評価としてのPSNRを参考に,.exrが使用できるように変更したつもりでした。
しかし、変更したプログラムを実行したところダメでした。
どのように直せばいいのでしょうか？

import cv2
import numpy as np
import math

#画像の読み込み
Original  = cv2.imread('test/Desk.exr') #元画像
Distorted = cv2.imread('out/000010_out.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)

を実行したところ

Traceback (most recent call last):
  File "PSNR.py", line 30, in <module>
    sumB += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 ) 
IndexError: index 1658880 is out of bounds for axis 0 with size 1658880

となってしまいました。

IndexError: index 1658880 is out of bounds for axis 0 with size 1658880
と検索したところ、
Numpyで行列の範囲外を切り抜くとどうなるかでは範囲外である。
IndexError: index is out of bounds for axis 0 with size 3 エラーの場合の対処では

原因

配列のサイズを超えて代入を行おうとしました。

対策

配列のサイズと代入するときの添え字を確認します。

となっていました。
自分のプログラムのどこを直せばいいのか、全くわからないので教えていただきたいです。

Comments

[jbpb0]

「#差の2乗の総和を計算」のすぐ上に、下記を追加して実行したら、何て表示されますか？

print(pixel_value_Ori.shape)
print(pixel_value_Dis.shape)
print(addr)

[退会済みユーザー]

$ python3 PSNR.py
(1688568,)
(1658880,)
1688568
Traceback (most recent call last):
File "PSNR.py", line 36, in <module>
sumB += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 )
IndexError: index 1658880 is out of bounds for axis 0 with size 1658880

となりました。

[jbpb0]

それを見て、何が悪いか考えてください

[退会済みユーザー]

pixel_value_Dis.shapeに
with size 1658880　　配列に限界の1658880
がはいっているのが悪いんですよね。

これをどうしたらいいのですか？

[jbpb0]

同じ場所に下記を追加して実行して、その結果表示を見比べてください

print(Original.shape)
print(Distorted.shape)

次に、この質問のコードでやろうとしてることができるか、上記の結果を加味しながら考えてください

[退会済みユーザー]

pythonで値の上限みたいなのが出たのが初めてなので、なぜこれが出てくるのかがわかりません。
どうか教えてください。

[退会済みユーザー]

(1688568,)
(1658880,)
1688568
(874, 644, 3)
(864, 640, 3)
Traceback (most recent call last):
File "PSNR.py", line 38, in <module>
sumB += pow ( (pixel_value_Ori[i]-pixel_value_Dis[i]), 2 )
IndexError: index 1658880 is out of bounds for axis 0 with size 1658880

実行した結果このようになりました。

> 次に、この質問のコードでやろうとしてることができるか、上記の結果を加味しながら考えてください
教えていただいた通り考えたところ、画像サイズが違うことに気が付きました。

もしかしなくても、画像サイズが違うからでしょうか？

[jbpb0]

> 画像サイズが違うからでしょうか？

そうです
この質問のコードでやろうとしてることは、二つの画像の同じ場所の画素の値が近いかどうかを、数値評価しようとしてるのですよね
その場合の前提として、画像のサイズ(縦横の画素数)が同じじゃないと無理ですよね

[退会済みユーザー]

なるほど、ありがとうございます。
なぜあのエラー分だけでわかったのか、教えていただけませんか？
あと、一つ質問があるので、こちらも教えていただきたいです。

サイトには
扱う画像は8[bit]であるとする. このとき, 画素値は0から255の値をとるので, 画素値の最大値は255となる.
ただし, Colorは画素を構成している色の数であるので3(i = 0でred, i=1でgreen, i = 2でblueとする), Nは全画素数, E(i,j)は2枚の画像間の, j番目の画素の色iにおける画素値の差とする.
とありました。
.exrの画素値はどうしたらいいのか？です。
そのまま使ってもいいものなのかわからなくなったので、教えていただきたく存じます。

$ sips -g all 000010_out.exr
/Users/1831083/hdrcnn/hdrcnn-master/out/000010_out.exr
pixelWidth: 640
pixelHeight: 864
typeIdentifier: com.ilm.openexr-image
format: exr
formatOptions: default
dpiWidth: 72.000
dpiHeight: 72.000
samplesPerPixel: 4
bitsPerSample: 16
hasAlpha: yes
space: RGB
profile: sRGB IEC61966-2.1 Linear

と実行したところ、samplesPerPixel: 4 bitsPerSample: 16となったので、
４チャンネルあり、１チャンネル16bitということだと思うのです。

あと、よろしければ回答をお願いしたく思います。

[jbpb0]

> なぜあのエラー分だけでわかったのか

「for i in range(addr):」で「i」は 0〜addr-1 まで順番に変わり、ループの中でその各「i」で「pixel_value_Ori[i]」と「pixel_value_Dis[i]」にアクセスしてますよね
配列のサイズよりも大きい添字でアクセスしてるエラーなので、「pixel_value_Ori[i]」と「pixel_value_Dis[i]」の少なくともどちらかのサイズは、addrよりも小さいはずです

最初のコード追加でそれを確認したら、「pixel_value_Ori[i]」のサイズはaddrと合ってましたが、「pixel_value_Dis[i]」のサイズはaddrよりも小さかった
なので、それらを作る元の画像のサイズが違う、と分かりますよね

Answer 1

「#差の2乗の総和を計算」のすぐ上に下記を追加して実行して、比較しようとしてる画像のサイズ(縦横の画素数とチャンネル数)が等しいかを確認してみてください

print(Original.shape)
print(Distorted.shape)

二つの画像の同じ場所の画素の値が近いかどうかを数値評価しようとしてるのだから、画像のサイズが同じじゃないと無理です

Answer 2

IndexError: index 1658880 is out of bounds for axis 0 with size 1658880

1658880のサイズの配列に、1658880でアクセスしようとしたのでサイズオーバー、とおっしゃってます
1658880のサイズの配列の範囲は、0から1658879までです

エラーが出ている行のリスト（配列）のサイズと添字の数値を確認しよう

Comments

[退会済みユーザー]

申し訳ございません。
それは、わかっているのですがどこを直せばいいのか、てんでわかりません。