【发布时间】:2017-11-18 06:29:56
【问题描述】:
即使我使用相同的数据集,我的训练分类也很高,但验证分类却很低。此问题仅在使用批量标准化时发生。我是否正确实施它?
使用批量标准化的代码:
train_datagen = ImageDataGenerator(rescale=1. / 255)
train_generator = train_datagen.flow_from_directory(
directory = '../ImageFilter/Images/',
target_size=(img_rows, img_cols),
batch_size=batch_size,
class_mode='categorical',
shuffle=True)
model = Sequential()
model.add(Convolution2D(16,
kernel_size=(3, 3),
strides=(2,2),
activation='relu',
input_shape=(img_rows, img_cols, 3)))
model.add(BatchNormalization())
model.add(MaxPooling2D((2,2), strides=(2,2)))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dense(2, activation='softmax'))
model.compile(loss='categorical_crossentropy',
optimizer='adam',
metrics = ['accuracy'])
epochs = 100
patience = 6
n_images = 91
file_path = 'imageFilterCNN.hdf5'
checkpointer = ModelCheckpoint(file_path, monitor='val_acc', verbose=0, save_best_only=True)
earlystop = EarlyStopping(monitor='val_acc', patience=patience, verbose=0, mode='auto')
tboard = TensorBoard('./logs')
model.fit_generator(
train_generator,
steps_per_epoch=n_images// batch_size,
epochs=epochs,
callbacks=[checkpointer, earlystop, tboard],
validation_data=train_generator,
validation_steps=n_images// batch_size)
输出: 时代 15/100 11/11 [===============================] - 2s - 损失:0.0092 - acc: 1.0000 - val_loss:3.0321 - val_acc:0.5568
【问题讨论】:
-
这些结果有什么奇怪的地方?训练准确率永远比测试好;你有什么理由期望泛化很简单?
-
我正在对其进行训练的同一数据集上进行测试。所以结果应该是非常相似的。
标签: machine-learning tensorflow keras