AttributeError: 'PngImageFile' object has no attribute 'shape'

Question

Python 初心者です。
２つの同じサイズの単色画像を受け取り、二乗平均誤差を返す関数を作ろうとしています。

from PIL import Image
import numpy as np
from gaussian_noise import add_gaussian_noise
import matplotlib.pyplot as plt

# 元となる画像の読み込み
im = Image.open("../images/lena-gray.png") 
plt.show(im)

# オリジナル画像の幅と高さを取得
width, height = im.size
#print(im.shape)

# オリジナル画像に平均値0、標準偏差10の画像を生成する
#im2 = Image.new(add_gaussian_noise(im, 0, 10), (width, height))
im2 = add_gaussian_noise(im, 0, 10)

def mse(im, im2):
    im  = np.array(im, dtype=np.int)
    im2 = np.array(im2, dtype=np.int)
    mse = np.mean((im-im2)**2)
    return mse

print(mse)

なお、add_gaussian_noiseについては、

import numpy as np

def add_gaussian_noise(src, mean=0, sigma = 10):
    noise = np.random.normal(mean, sigma, src.shape)
    dst = np.clip(src + noise, 0., 255.)
    return dst.astype(src.dtype)

です。

実行すると、

File "D:\WPy64-3760\notebooks\chap3\gaussian_noise.py", line 4, in add_gaussian_noise
noise = np.random.normal(mean, sigma, src.shape)

AttributeError: 'PngImageFile' object has no attribute 'shape'
と出てきてしまうのですが、
PNGにshapeが無いといわれてもピンときません。
画像ファイルは存在している筈(存在はしている)なのですが…

お力添え、よろしくお願いします。

Answer 1

shapeというのはnumpy配列などのクラスに存在する属性で、Image.openが返すPngImageFileクラスのオブジェクトには存在しません。

クラスなどの概念がわからない場合、まずオブジェクト指向について学んでみてください。

shape属性を使いたい場合はnumpy配列に変換が必要です。sizeでとっても良いのですが、numpyのshapeとは縦横が反転するので注意が必要になります。

Comments

[tfulcrum]

ありがとうございました。PngImageFileがクラスという事に気づきませんで、つまり、Imge.openを使った場合にこう出る、という事に気づいていませんでした。
im=np.asarray(Image.open("../images/lena-gray.png"))
として
print(mse(im, im2))
としたところ、
97.78177…
と出ました。
もっとも、違う値が出る様な気もしますが…

[hayataka2049]

元画像と元画像にノイズをまぶしたものの間でmseを計算しているので、雑に言えばノイズの標本分散を求めていることになります。もっとも上下をクリップさせているので、たぶん少し小さめの数字にはなるでしょう。
標準偏差10のノイズを足して97.78177…になったなら「極めてまっとうな結果」だと思います。

[tfulcrum]

回答に対する補足、ありがとうございました。
まさにざっくりいえばそうでした。
ただ、値が変わってもおかしくない…という発想はありませんでした。