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

Question

我正在尝试可视化 MobileNet 中卷积层的输入和输出。TensorFlow 中是否有任何功能或工具可以让我们查看每一层的输入和输出的确切值？所以假设我有一张图像，我正在做物体检测。模型跑了一堆层，怎么看数据流？

Answer 1

我会看看TensorFlow 调试器 tfdbg。您可以检查中间激活，包括梯度。

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

第三种选择是向tf.Print图中添加节点。同样，您可以添加摘要并将其发送到 TensorBoard。