【发布时间】:2018-04-12 00:45:19
【问题描述】:
我正在尝试可视化 MobileNet 中卷积层的输入和输出。 TensorFlow 中是否有任何功能或工具可以让我们查看每一层的输入和输出的确切值? 所以假设我有一张图像,我正在做物体检测。模型跑了一堆层,怎么看数据流?
【问题讨论】:
标签: tensorflow convolutional-neural-network
我正在尝试可视化 MobileNet 中卷积层的输入和输出。 TensorFlow 中是否有任何功能或工具可以让我们查看每一层的输入和输出的确切值? 所以假设我有一张图像,我正在做物体检测。模型跑了一堆层,怎么看数据流?
【问题讨论】:
标签: tensorflow convolutional-neural-network
我会看看the TensorFlow debugger, tfdbg。您可以检查中间激活,包括梯度。
启用eager execution 是另一种选择,在这种情况下,TensorFlow 的行为更像 numpy,而不是特定领域的语言。您可以使用常规打印语句打印中间激活,或使用 Python 的 pdb。
第三种选择是将tf.Print 节点添加到图中。同样你可以add summaries and send them to TensorBoard。
【讨论】: