Как использовать Skimage для изменения размера тренировочных данных?

Question

Я пытаюсь понять, как работает функция изменения размера в Skimage. Я пытаюсь получить данные x_train, представляющие собой набор данных CIFAR10 с формой x_train shape: (50000, 32, 32, 3), для изменения размера (244, 244, 3) для использования в модели VGG19. Я пытался использовать следующий код для изменения данных, но не уверен, что результаты верны, так как попытка сделать plt.imshow (resized_xtrain [0]) дает мне странное изображение.

from skimage import transform
for i in range (50000):
    resized_xtrain= transform.resize(x_train[i], (224, 224, 3), order=1, mode='reflect')

Когда я делаю resized_xtrain.shape, я получаю (244, 244, 3). Я не совсем понимаю, как настроить это так, чтобы все 50 000 тренировочных образов были изменены и установлены на resized_xtrain. Как мне изменить размер, чтобы поместить все тренировочные данные в массив?

Answer 1

Вы забыли проиндексировать свой resized_xtrain, сделайте это следующим образом:

from skimage import transform

resized_xtrain = np.zeros((50000,224,224,3))
for i in range (50000):
    resized_xtrain[i] = transform.resize(x_train[i], (224, 224, 3), order=1, mode='reflect')