我正在尝试将使用 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 中看到正确的预期节点?