ImageDataGenerator - 使用 model.fit 而不是 model.fit_generator 进行训练
ImageDataGenerator - trained with model.fit instead of model.fit_generator
我是使用 Keras 的 ImageDataGenerator 的初学者,我不小心使用了 model.fit 而不是 model.fit_generator。
train_gen = gen_Image_data()
test_gen = ImageDataGenerator()
train_samples = train_gen.flow(X,y, batch_size=64)
test_samples = test_gen.flow(X_val, y_val, batch_size=64)
history = model.fit(train_samples, steps_per_epoch = np.ceil(len(X)/64),
validation_data=(test_samples),
validation_steps=np.ceil(len(X_val)/64),
epochs=300, verbose=1, callbacks=[es])
这是一个明显的错误吗,我是否必须用 fit_generator 重新训练所有内容?
感谢大家的帮助
更新 我忘记了 gen_Image_data()
的代码
def gen_Image_data():
gen = ImageDataGenerator(
width_shift_range=0.1,
horizontal_flip=True)
return gen
您无需重新训练模型,因为 model.fit
方法也支持生成器,而且 model.fit_generator
也包含在 model.fit
方法中!
我是使用 Keras 的 ImageDataGenerator 的初学者,我不小心使用了 model.fit 而不是 model.fit_generator。
train_gen = gen_Image_data()
test_gen = ImageDataGenerator()
train_samples = train_gen.flow(X,y, batch_size=64)
test_samples = test_gen.flow(X_val, y_val, batch_size=64)
history = model.fit(train_samples, steps_per_epoch = np.ceil(len(X)/64),
validation_data=(test_samples),
validation_steps=np.ceil(len(X_val)/64),
epochs=300, verbose=1, callbacks=[es])
这是一个明显的错误吗,我是否必须用 fit_generator 重新训练所有内容?
感谢大家的帮助
更新 我忘记了 gen_Image_data()
def gen_Image_data():
gen = ImageDataGenerator(
width_shift_range=0.1,
horizontal_flip=True)
return gen
您无需重新训练模型,因为 model.fit
方法也支持生成器,而且 model.fit_generator
也包含在 model.fit
方法中!