我创建了Keras多输出模型(实际上它有两个输出)。
另外,我有一个精心挑选的图像数据集(一个训练数据目录,每个图像类包含一个子目录,并填充有.png或.jpg图像),我想在训练期间使用{的flow_from_directory
方法传递给模型{1}}类。但是,默认情况下,此类返回一个ImageDataGenerator
产生的元组DirectoryIterator
。
如何创建自定义生成器,该生成器将使用(x, y)
方法以[input_x], [logits_variance,softmax]
格式返回生成器对象?
我提供了模型结构的屏幕截图,以便更好地理解我的意思。