python输出CNN卷积神经网络 某一层输出数据的方法总结
本人在搭建CNN神经网络时,发现了两种可以输出某一层的方法。第一种:functor = tf.keras.backend.function([model.layers[0].input], [model.layers[-1].output])predict=functor(trainX )在该段代码中,第一个参数是模型的输入部分,第二个参数是最后一层的输出(可自行选择哪一层),调用该函数时,传入你的数据trainX,最后predict返回的就是最后一层的输出。第二种:functo
原创
2021-07-29 14:53:47 ·
4648 阅读 ·
4 评论