有了这个生成器,我们就可以用fit_generator 方法进行训练,格式套路如下:
model.fit_generator(generator,
steps_per_epoch=...,
epochs=...,
verbose=...,
callbacks=...,
validation_data=...,
validation_steps=...,
validation_freq=...,
class_weight=None=...,
max_queue_size=...
workers=...,
use_multiprocessing=...,
)
除此以外我们还可以搞批量预测:
model.predict_generator()
Keras用动态数据生成器(DataGenerator)和fitgenerator动态训练模型 - Freiburger - 博客园