我使用以下代码使用发票图像训练 CNN 模型。
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_data_dir,
target_size =(img_width, img_height),
batch_size = batch_size)
validation_generator = test_datagen.flow_from_directory(
validation_data_dir,
target_size =(img_width, img_height),
batch_size = batch_size)
model.fit_generator(train_generator,
steps_per_epoch = nb_train_samples // batch_size,
epochs = epochs, validation_data = validation_generator,
validation_steps = nb_validation_samples // batch_size)
Run Code Online (Sandbox Code Playgroud)
问题是我在训练数据集中只使用了直立图像。我所有的图像都如下图所示:
训练结束后,当我想发送如下所示的图像时,我的模型无法预测其正确的类别。
如下所示,我将horizontal_flip = True发送到ImageDataGenerator
train_datagen = ImageDataGenerator(
rescale = 1. / 255,
shear_range = …Run Code Online (Sandbox Code Playgroud) python image-processing conv-neural-network keras tensorflow