gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-02 04:28:40 25 4
gpt4 key购买 nike

现在,我能够在我的图像数据上训练和测试神经网络。我希望能够从 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()

最佳答案

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

关于python - 如何从 Keras 中的 ImageDataGenerator 格式化 X 和 Y 数据?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50992702/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com