【525】keras.utils.Sequence 制作生成集 - 在 fit() 中使用
如何使用Keras fit和fit_generator(动手教程)
自定义 数据生成器
def batch_image_generator(input_img_paths, target_img_paths_louding, target_img_paths_louti): """Returns tuple (input, target) correspond to batch #idx.""" for idx in range(len(input_img_paths)//batch_size): x = np.zeros((batch_size,) + img_size + (3,), dtype="float32") ... y1 = np.zeros((batch_size,) + img_size + (1,), dtype="float32") ... y2 = np.zeros((batch_size,) + img_size + (1,), dtype="float32") ... # 使用字典形式的话,需要保证 key值 的一一对应关系 x = {'inputs': x} y = {'outputs1': y1, 'outputs2': y2} yield (x, y)
之后调用 fit_generator
train_gen = batch_image_generator(...) model.fit_generator(train_gen, ...)