(ValueError: setting an array element with a sequence.)の原因と対策方法を教えてください

Question

質問失礼いたします。初歩的な質問で大変恐縮なのですが、お答え頂ければ幸いです。

https://aiacademy.jp/texts/show/?id=164
こちらのサイトを参考にプログラムしているのですが、検索してもエラーの原因がわかりませんでした。
原因・対策を教えてください。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "c:\Users\----\Desktop\SS\createNPY.py", line 77, in <module>
    np.save("./dataset.npy", xy)
  File "<__array_function__ internals>", line 200, in save
  File "C:\Users\----\AppData\Roaming\Python\Python311\site-packages\numpy\lib\npyio.py", line 521, in save
    arr = np.asanyarray(arr)
          ^^^^^^^^^^^^^^^^^^
ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape after 1 dimensions. The detected shape was (4,) + inhomogeneous part.

該当のソースコード

from PIL import Image
import os, glob
import numpy as np
from PIL import ImageFile
# IOError: image file is truncated (0 bytes not processed)回避のため
ImageFile.LOAD_TRUNCATED_IMAGES = True

# indexを教師ラベルとして割り当てるため、0にはdogを指定し、1には猫を指定
classes = ["circle", "square"]
num_classes = len(classes)
image_size = 64
num_testdata = 25

X_train = []
X_test  = []
y_train = []
y_test  = []

for index, classlabel in enumerate(classes):
    photos_dir = "./data/" + classlabel
    files = glob.glob(photos_dir + "/*.jpg")
    for i, file in enumerate(files):
        image = Image.open(file)
        image = image.convert("RGB")
        image = image.resize((image_size, image_size))
        data = np.asarray(image)
        if i < num_testdata:
            X_test.append(data)
            y_test.append(index)
        else:

            # angleに代入される値
            # -20
            # -15
            # -10
            #  -5
            # 0
            # 5
            # 10
            # 15
            # 画像を5度ずつ回転
            for angle in range(-20, 20, 5):

                img_r = image.rotate(angle)
                data = np.asarray(img_r)
                X_train.append(data)
                y_train.append(index)
                # FLIP_LEFT_RIGHT　は 左右反転
                img_trains = img_r.transpose(Image.FLIP_LEFT_RIGHT)
                data = np.asarray(img_trains)
                X_train.append(data)
                y_train.append(index) # indexを教師ラベルとして割り当てるため、0にはdogを指定し、1には猫を指定

X_train = np.array(X_train)
X_test  = np.array(X_test)
y_train = np.array(y_train)
y_test  = np.array(y_test)

"""
print(X_train.ndim)
print(X_test.ndim)
print(y_train.ndim)
print(y_test.ndim)

print(X_train.size)
print(X_test.size)
print(y_train.size)
print(y_test.size)

print(X_train.shape)
print(X_test.shape)
print(y_train.shape)
print(y_test.shape)
"""

xy = (X_train, X_test, y_train, y_test)
np.save("./dataset.npy", xy)

試したこと

下のコードを入れて次元と要素数を取得しました

print(X_train.ndim)
print(X_test.ndim)
print(y_train.ndim)
print(y_test.ndim)

print(X_train.size)
print(X_test.size)
print(y_train.size)
print(y_test.size)

print(X_train.shape)
print(X_test.shape)
print(y_train.shape)
print(y_test.shape)

結果↓

4
4
1
1
3922329600
614400
319200
50
(319200, 64, 64, 3)
(50, 64, 64, 3)
(319200,)
(50,)

Comments

[meg_]

エラーメッセージには「File "c:\Users\----\Desktop\SS\createNPY.py", line 77, in <module>」とありますが、該当のソースコードは60行までしかありません。該当のソースコードを実行しても同じエラーが発生しますか？

[hk1217_]

すいません！
print(X_train.ndim)
print(X_test.ndim)
print(y_train.ndim)
print(y_test.ndim)

print(X_train.size)
print(X_test.size)
print(y_train.size)
print(y_test.size)

print(X_train.shape)
print(X_test.shape)
print(y_train.shape)
print(y_test.shape)
このコードをコメントアウトしていたので行数に狂いがありました

[hk1217_]

更新させていただきました

[meg_]

下記コードは成功しますか？
np.save("./test.npy", X_train)

Answer 1

参考サイトのコードでsaveを使っているのは解せませんが、かわりにsavez_compressedだと複数のshapeの異なるarrayを圧縮保存できます。
読み込みも以下のような感じでloadできます。
参考：numpy.savez_compressed：複数のNumPy配列を圧縮＆バイナリ保存

from PIL import Image
import numpy as np


def disp(*xy):
    for e in xy:
        print(e.shape)

image_size = 64

X_train = []
X_test  = []
y_train = []
y_test  = []

image = Image.new('RGB',(image_size,image_size))
data = np.asarray(image)

for i in range(2):
    X_test.append(data)
    y_test.append(0)

for i in range(10):
    X_train.append(data)
    y_train.append(0)

X_train = np.array(X_train)
X_test  = np.array(X_test)
y_train = np.array(y_train)
y_test  = np.array(y_test)

xy = (X_train, X_test, y_train, y_test)
disp(X_train, X_test, y_train, y_test)

np.savez_compressed('ret.npz', *xy)

# 読み込み
loaded = np.load('ret.npz')
X_train, X_test, y_train, y_test = loaded.values()
disp(X_train, X_test, y_train, y_test)
"""
(10, 64, 64, 3)
(2, 64, 64, 3)
(10,)
(2,)
"""