具有多个输出的Imagedatagenerator .flowfromDataframe

时间:2020-09-26 23:04:11

标签: tensorflow keras conv-neural-network

我尝试用一​​个图像作为输入和4个输出来实现一个模型。 该数据被标记为[路径,y1,y2,y3,y4]。 y1是分类分类,y2-4是二进制分类。我应该选择哪种课堂模式?我尝试使用“原始”,因为我只有数字值。

我想我必须编写自己的生成器,对吗? 如果我查看从结构生成的图,模型本身看起来还不错。 唯一的问题是产生模型的标签。我尝试了这个自定义生成功能

def multiple_outputs(generator, dataframe, batch_size):
    gen = generator.flow_from_dataframe(
        dataframe = dataframe,
        x_col="path",
        y_col=["y1", "y2", "y3", "y4"]
        target_size=(30,30),
        batch_size=batch_size
    ) 

    while True:
        gnext = gen.next()
        yield (gnext[0], [gnext[1], gnext[1], gnext[1], gnext[1]])

非常感谢您的帮助, 戴夫

0 个答案:

没有答案