我尝试用一个图像作为输入和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]])
非常感谢您的帮助, 戴夫