1

我正在尝试将使用 Tensorflow 中的 Keras API 训练的自定义模型部署到 google aiy vision kit v2,当.pb使用我得到的视觉套件编译器从 Tensorflow 文件编译模型时:

output node named "<output node>" does not appear in the graph.

我的问题是,如何从 tensorboard 中获取参数中传递给编译器的输出节点--output_tensor_names

我已经尝试过返回的值:

keras_model.outputs[0].op.name    # returns final_output/Sigmoid
keras_model.outputs[0].name    # returns final_output/Sigmoid:0

final_output,这是 keras 中最后一层的名称。

但是这些都不起作用,我怎样才能在 tensorboard 中看到正确的预期节点?

4

0 回答 0