时间戳

首页 > 解决方案 > 如何在 Keras 中格式化 ImageDataGenerator 的 X 和 Y 数据？

python - 如何在 Keras 中格式化 ImageDataGenerator 的 X 和 Y 数据？

问题描述

现在，我能够在我的图像数据上训练和测试神经网络。我希望能够从 DirectoryIterator train_generator 中提取图像数据和相应的标签，这样我就可以对这些数据进行洗牌并执行交叉验证。有没有办法提取（如果需要也可以格式化）这些数据？

from keras.preprocessing.image import ImageDataGenerator

img_width, img_height = 150, 150
train_data_dir = '/train'
datagen = ImageDataGenerator(rescale=1./255) #rescales [0,1]
train_generator = datagen.flow_from_directory(train_data_dir, target_size=(img_width, img_height), shuffle=True, batch_size=32, class_mode='binary')
#I want to do something like (X, Y) = train_generator.getData()

标签： pythonkeras

解决方案

train_generator = datagen.flow_from_directory(
   train_data_dir, 
   target_size=(img_width, img_height), 
   shuffle=True, 
   batch_size=SIZE_OF_DATASET,  ## change to size of entire dataset
   class_mode='binary')

for X, Y in train_generator:
   break

推荐阅读