Как перебрать набор данных для изменения размера каждого изображения? - PullRequest
Новая
       14

Как перебрать набор данных для изменения размера каждого изображения?

0 голосов
/ 14 апреля 2019

Я использую набор данных Cifar10 для запуска в программе обучения передаче.Я застрял при попытке получить данные x_train, чтобы иметь правильную форму, которая необходима для программы.Cifar10 дает x_train как (50,000 , 32, 32, 3) ... Моя цель состоит в том, чтобы превратить эти x_train_resized данные в (50,000, 224, 224, 3)

Я не уверен, как циклически проходить данные / если они будут такими же, какMatlab, с которым я знаком.То, что я пробовал, это: 

from skimage import transform
from keras.models import Model


model2 = Model(inputs=base_model.input, outputs=base_model.get_layer('fc2').output)
results=[]
resized_xtrain=[]
for i in range (50000):
    resized_xtrain= transform.resize(x_train[i], (224, 224, 3), order=1, mode='reflect')

    x = np.expand_dims(resized_xtrain, axis=0)
    x = preprocess_input(x)

    fc2_features = model2.predict(x)
    fc2_features.shape
    results.append(fc2_features)

Опять же, я не уверен, как получить данные размером 50 000 изображений.когда я запускаю этот код, я получаю 224, 224, 3, но не 50 000.Любая помощь будет отличной!Спасибо.

...