我曾经手动制作它,但现在我使用 flow_from_directory 用我自己的数据训练我的网络。我只有一个问题。当我进行 model.predict() 时,我如何知道我的预测索引 0 是针对标签类别狗,而索引 1 是针对类别猫?
我正在使用的代码如下。
train_datagen = ImageDataGenerator(
rescale=1./255,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True)
test_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
train_images_path,
target_size=(64, 64),
batch_size=batch_size)
validation_generator = test_datagen.flow_from_directory(
validate_images_path,
target_size=(64, 64),
batch_size=batch_size)
early_stopping = keras.callbacks.EarlyStopping(monitor='val_acc', min_delta=0, patience=3, verbose=1, mode='auto')
history = model.fit_generator(
train_generator,
steps_per_epoch=1700,
epochs=epochs,
verbose=1,
callbacks=[early_stopping],
validation_data=validation_generator,
validation_steps=196
)
我想知道的是图像对与真实标签。
谢谢
您可以使用 class_indices 属性获得生成器生成的每个类的索引。
print(validation_generator.class_indices)
简单的...
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)