Keras:如何将ImageDataGenerator类的flow_from_directory方法用于多输出模型

时间:2020-01-24 13:56:08

标签: python keras

我创建了Keras多输出模型(实际上它有两个输出)。 另外,我有一个精心挑选的图像数据集(一个训练数据目录,每个图像类包含一个子目录,并填充有.png或.jpg图像),我想在训练期间使用{的flow_from_directory方法传递给模型{1}}类。但是,默认情况下,此类返回一个ImageDataGenerator产生的元组DirectoryIterator

如何创建自定义生成器,该生成器将使用(x, y)方法以[input_x], [logits_variance,softmax]格式返回生成器对象?

我提供了模型结构的屏幕截图,以便更好地理解我的意思。

enter image description here

0 个答案:

没有答案