我想保存一些DNN中间层产生的特征Tensor,并将其用于另一个训练网络。例如,我将一些输入提取到alexnet,并获得中间层特征向量,并将其保存在一个文件中。然后,我想将这些Tensor用作另一个网络的输入。我如何保存它们?我想保存它们,以便我可以逐个加载它们。谢谢。
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] )
有关详细信息,请参阅本文档。谢谢。
1条答案
按热度按时间qyzbxkaa1#
您可以使用下面的代码提取Sequential模型中所有中间层的输出。
有关详细信息,请参阅本文档。谢谢。