Невозможно изменить размер изображения с помощью cv2

Question

Я пытаюсь изменить размер изображения cifar10 с 32x32 до 96x96.

(train_images, train_labels), (test_images, test_labels) = cifar10.load_data()

train_images_reshaped = np.array((50000, 96, 96, 3,))
for a in range(len(train_images)):
    train_images_reshaped[a] = cv2.resize(train_images[a], dsize=(96, 96), interpolation=cv2.INTER_CUBIC)

Но получаю ошибку

ValueError: setting an array element with a sequence.

Что происходит не так? Есть ли какие-либо альтернативы для достижения моей цели?

Answer 1

Я думаю, что вы хотели сделать

train_images_reshaped = np.zeros((50000, 96, 96, 3,))

вместо

train_images_reshaped = np.array((50000, 96, 96, 3,))

Answer 2

Вы инициализируете train_images_reshaped неправильно, это ndarray формы (50000, 96, 96, 3,), а не массив (50000, 96, 96, 3,). Попробуйте это:

train_images_reshaped = np.zeros_like(train_images, shape=(50000, 96, 96, 3,))