python - keras 中中间层的输出作为输入传递一个 DirectoryIterator

Question

有没有办法在 Keras 中通过 DirectoryIterator 作为输入来获取中间层的输出？

我已经阅读了与这个和这个问题相关的 Keras 文档，但据我所知，这些方法使用单个样本作为输入。

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

我使用 TensorFlow 作为后端。

标签： pythontensorflowdeep-learningkeras

Answer 1

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

然后使用partialModeltofit或predict与目录迭代器一起使用。