如何打印胶囊层的输出矢量？

Question

我和胶囊网络一起工作。我的参考论文是这里。本文阐述了一次胶囊层产生位姿参数的8D矢量。我的问题是，我们能从图层的输出得到矢量并打印出8D矢量吗？

Answer 1

由于问题被标记为keras，所以我假设模型是使用keras库构建的。

我们能从图层的输出得到矢量并打印出8D矢量吗？

是的你可以。首先，您必须访问模型的输入和胶囊层的输出。

inputs = model.input
outputs = model.layers[index_of_the_caps_layer].output

Keras提供了运行计算图的keras.backend.function。您需要传递模型输入占位符的numpy数组。

import keras.backend as K 
input_values # Input values to the model 
function = K.function([inputs, K.learning_phase()], outputs )
caps_output = function([input_values, 1.0])
print(caps_output)