python输出CNN卷积神经网络 某一层输出数据的方法总结

本人在搭建CNN神经网络时，发现了两种可以输出某一层的方法。

第一种：

functor = tf.keras.backend.function([model.layers[0].input],
 [model.layers[-1].output])
predict=functor(trainX )

在该段代码中，第一个参数是模型的输入部分，第二个参数是最后一层的输出(可自行选择哪一层)，调用该函数时，传入你的数据trainX，最后predict返回的就是最后一层的输出。

第二种：

functor=tf.keras.models.Model(inputs=model.input,outputs=model.layers[-1].output)
predict= layer_model.predict(trainX)

在该段代码中， 和上面的比较类似，也是需要输入两个参数，不过使用的predict函数进行预测。