如何使用Keras model.evaluate_generator

时间:2017-03-15 13:19:12

标签: theano keras

我已使用以下数据扩充来节省内存。

total number of test images = 400
batch size = 128

当我使用model.evaluate_generator检查测试集的准确性时,它与我模型的上一个时期的最终validation_accuracy不同。 此外,当我重复此操作时,model.evaluate_generator的输出会发生变化。

下面是我的代码。

请帮忙!

train_datagen = ImageDataGenerator(
        rescale=1./255,)

test_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_datagen.flow_from_directory(
        train_data_dir,
        color_mode= "grayscale",
        target_size=(img_width, img_height),
        batch_size=128,
        class_mode='categorical',)    

validation_generator = test_datagen.flow_from_directory(
        validation_data_dir,
        color_mode= "grayscale",
        target_size=(img_width, img_height),
        batch_size=128,
        class_mode='categorical')

#%%
hist = model.fit_generator(
        train_generator,
        samples_per_epoch=nb_train_samples,
        nb_epoch=nb_epoch,
        validation_data=validation_generator,
        nb_val_samples=nb_validation_samples)

scoreSeg = model.evaluate_generator(validation_generator, 400)
print("Accuracy = ",scoreSeg[1])

0 个答案:

没有答案