如何在MobileNet中查看卷积层的输入和输出的确切值

时间:2018-04-09 05:28:23

标签: tensorflow convolutional-neural-network

我正在尝试可视化MobileNet中卷积层的输入和输出。 TensorFlow中是否有任何功能或工具可以让我们看到输入的确切值,每层的输出? 所以,假设我有一个图像,我正在进行物体检测。该模型贯穿一堆层,如何查看数据流?

1 个答案:

答案 0 :(得分:1)

我会看看the TensorFlow debugger, tfdbg。您可以检查中间激活,包括渐变。

启用eager execution是另一种选择,其中TensorFlow的行为更像numpy而不是特定于域的语言。您可以使用常规打印语句打印中间激活,或使用Python的pdb。

第三个选项是向图表添加tf.Print个节点。同样,您可以add summaries and send them to TensorBoard