alex_bn_lee

导航

【639】keras 中 fit_generator 的 数据生成器

【525】keras.utils.Sequence 制作生成集 - 在 fit() 中使用 

如何使用Keras fit和fit_generator(动手教程)

keras:怎样使用 fit_generator 来训练多个不同类型的输出【推荐】

Keras多输出(多任务)如何设置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, ...)

 

posted on 2021-08-09 17:33  McDelfino  阅读(356)  评论(0编辑  收藏  举报