我正在尝试可视化MobileNet中卷积层的输入和输出。 TensorFlow中是否有任何功能或工具可以让我们看到输入的确切值,每层的输出? 所以,假设我有一个图像,我正在进行物体检测。该模型贯穿一堆层,如何查看数据流?
答案 0 :(得分:1)
我会看看the TensorFlow debugger, tfdbg。您可以检查中间激活,包括渐变。
启用eager execution是另一种选择,其中TensorFlow的行为更像numpy而不是特定于域的语言。您可以使用常规打印语句打印中间激活,或使用Python的pdb。
第三个选项是向图表添加tf.Print
个节点。同样,您可以add summaries and send them to TensorBoard。