keras 在文件中保存多个Tensor

gojuced7  于 2022-11-13  发布在  其他
关注(0)|答案(1)|浏览(157)

我想保存一些DNN中间层产生的特征Tensor,并将其用于另一个训练网络。例如,我将一些输入提取到alexnet,并获得中间层特征向量,并将其保存在一个文件中。然后,我想将这些Tensor用作另一个网络的输入。我如何保存它们?我想保存它们,以便我可以逐个加载它们。谢谢。

qyzbxkaa

qyzbxkaa1#

您可以使用下面的代码提取Sequential模型中所有中间层的输出。

initial_model = keras.Sequential(
    [
        keras.Input(shape=(250, 250, 3)),
        layers.Conv2D(32, 5, strides=2, activation="relu"),
        layers.Conv2D(32, 3, activation="relu"),
        layers.Conv2D(32, 3, activation="relu"),
    ]
)
feature_extractor = keras.Model(
    outputs=[layer.output for layer in initial_model.layers]
)

有关详细信息,请参阅本文档。谢谢。

相关问题