【639】keras 中 fit_generator 的 数据生成器
【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, ...)
浙公网安备 33010602011771号