在keras中输出中间层作为DirectoryIterator的输入

Question

有没有办法让Keras的中间层输出传递DirectoryIterator作为输入？

我已经阅读了documentation of Keras related to this和this question但据我所知，这些方法使用单个样本作为输入。

我想这样做是因为它比逐个读取样本更简单，并且比将所有内容放在内存上的计算成本更低。

我正在使用TensorFlow作为后端。

Answer 1

desiredOutputs = model.get_layer('layer_name').outputs #or model.layers[n].outputs
partialModel = Model(model.inputs,desiredOutputs)

然后使用partialModel到fit或predict和目录迭代器。