【发布时间】:2017-06-14 04:36:51
【问题描述】:
我正在尝试显示卷积神经网络每一层的输出。 我使用的后端是 TensorFlow。 代码如下:
import ....
from keras import backend as K
model = Sequential()
model.add(Convolution2D(32, 3, 3, input_shape = (1,28,28)))
convout1 = Activation('relu')
model.add(convout1)
(X_train, y_train), (X_test, y_test) = mnist_dataset = mnist.load_data("mnist.pkl")
reshaped = X_train.reshape(X_train.shape[0], 1, X_train.shape[1], X_train.shape[2])
from random import randint
img_to_visualize = randint(0, len(X_train) - 1)
# Generate function to visualize first layer
# ERROR HERE
convout1_f = K.function([model.input(train=False)], convout1.get_output(train=False)) #ERROR HERE
convolutions = convout1_f(reshaped[img_to_visualize: img_to_visualize+1])
完整的错误是:
convout1_f = K.function([model.input(train=False)], convout1.get_output(train=False)) TypeError: 'Tensor' object is not 可调用
非常感谢任何评论或建议。谢谢。
【问题讨论】:
-
我的答案正确吗?你试过了吗?
标签: machine-learning tensorflow neural-network keras conv-neural-network