cannot reshape array of sizeというエラーについて

Question

import imageio
import numpy as np


PHOTO_SIZE = 150 * 150 * 3


X = np.empty((0, PHOTO_SIZE), np.uint8)
y = np.empty(0, np.uint8)


for i in range(1, 61):
    
   p1 = imageio.imread(f"./Downloads/hifu_photos/melano/melano-{i:03d}.jpg").reshape(1, 
   PHOTO_SIZE)

    X = np.append(X, p1, axis = 0)

    y = np.append(y, np.array([0], dtype = np.uint8))
    
    p2 = imageio.imread(f"./Downloads/hifu_photos/normal/normal-{i:03d}.jpg").reshape(1, 
    PHOTO_SIZE)

    X = np.append(X, p2, axis = 0)
    y = np.append(y, np.array([1], dtype = np.uint8))```

と入力すると
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-5-4cd188e682d7> in <module>
     18     y = np.append(y, np.array([0], dtype = np.uint8))
     19 
---> 20     p2 = imageio.imread(f"./Downloads/hifu_photos/normal/normal-{i:03d}.jpg").reshape(1, PHOTO_SIZE)
     21     X = np.append(X, p2, axis = 0)
     22     y = np.append(y, np.array([1], dtype = np.uint8))

ValueError: cannot reshape array of size 90000 into shape (1,67500)

と出てきてしまいます。

なぜ90000という数字が出てくるのでしょうか。。。
カラー画像で150*150*3=67500のはずなのですが。。。

Comments

[toast-uz]

コードは、```で囲ってください。
また、できたらエラーメッセージも同様にされるとわかりやすいです。

[yy-_.15]

承知致しました。
次回より気をつけたいと思います！

Answer 1

imageioのカラー画像は、透過率も含めた4次元データになっているようです。

参考: 【Python】Imageioの使い方

Comments

[yy-_.15]

なるほど！透過率という概念があるのですね！
参考サイトを一度見てみたいと思います！
ありがとうございます！

[yy-_.15]

RGBAを RGBに一括変換する方法などはありますでしょうか？
どこのサイトも制限がかかってしまっておりまして。。。

[toast-uz]

ちゃんと意味がある透明度だと変換できないです。背景画像との合成が必要ですので。
なんちゃって透明度であって、その情報がいらないのであれば、reshapeの前に[:,:,0:4]を入れて色データの4つ目の要素を削ってしまえばよいです。

[yy-_.15]

それでは透明度はそのままにしてコードの方を書き換えた方が良いということでしょうか？

[toast-uz]

透明度が必要なのであれば、そうでしょうね。
なお、p1の方の画像は、透明度データ無しだったのでしょう。

[yy-_.15]

ありがとうございます。
試してみます！

[yy-_.15]

RGBデータに変換するプログラムを書くことで
解決できました！

[toast-uz]

ほほう。単純にアルファ要素を削除するだけなのか、何か別の手法なのか、参考までに教えて頂けますか？

[yy-_.15]

お返事が遅れました！
https://rinsaka.com/python/tegaki-number-02.html
このサイトが参考になるかと思います！

[yy-_.15]

あと、透過率を含めたとしても四次元ではなく三次元のままではないでしょうか？

[toast-uz]

ありがとうございます。RGBA→RGBは、やはりAを単純削除ですね。
次元の言い方は、numpyとしては質問者様のご指摘通りです。3次元目がRGBAで大きさ4ということですね。一方RGBAの各数値を4次元空間の点としてみることもできますので、そちらの言い方をしました。紛らわしかったですね。