alex_bn_lee

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

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

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

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

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

Keras多输出(多任务)如何设置fit_generator

  自定义 数据生成器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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

1
2
train_gen = batch_image_generator(...)
model.fit_generator(train_gen, ...)

 

posted on   McDelfino  阅读(385)  评论(0编辑  收藏  举报

编辑推荐:
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
阅读排行:
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
· .NET10 - 预览版1新功能体验(一)
点击右上角即可分享
微信分享提示