mnistデータのresizeについて

Question

kerasでloadしたmnistデータをx,y共に2倍に拡大しようと試みていますが、エラーがでてうまくいきません。
試しに以下のコードを書いてみたところ、could not broadcast input arrat from (56, 56) into shape (28, 28)とエラーが出てきました。
どなたか解決方法を教えて頂きたいです。よろしくお願いします。

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train[0] = cv2.resize(x_train[0], (56, 56))

Answer 1

cv2.resize() ではリサイズはできています。
問題なのは、既存の要素にリサイズした画像を代入しようとしている以下の部分です。

x_train[0] = リサイズした (56, 56) の numpy 配列

x_train[0] は (28, 28) と確定しているので、ここに (56, 56) の numpy 配列を代入することはできません。
以下のように新しく配列を作り直してください。

import cv2
import numpy as np
from keras.datasets import mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
print(x_train.shape, x_test.shape)  # (60000, 28, 28) (10000, 28, 28)

x_train = np.array([cv2.resize(img, (56, 56)) for img in x_train])
x_test = np.array([cv2.resize(img, (56, 56)) for img in x_test])
print(x_train.shape, x_test.shape)  # (60000, 56, 56) (10000, 56, 56)

Comments

[Kohei_KESE]

無事解決することができました。一度定義した配列は再定義する必要があったのですね。
回答ありがとうございます。