tensorflow.js - 未能将 Tensorflow .pd 转换为 json

Question

尝试将保存的模型转换为 json 用于 tensorflow js。遵循来自https://github.com/tensorflow/tfjs/tree/master/tfjs-converter的示例

版本：tensorflowjs 1.2.9

依赖版本：keras 2.2.4-tf tensorflow 1.14.0

运行这个 cmd：

tensorflowjs_converter --input_format=tf_saved_model --output_format=tfjs_graph_model --signature_name=serving_default --saved_model_tags=serve /saved_model  /web_model

运行代码时出现此错误消息：

F .\tensorflow/core/grappler/graph_view.h:332] Check failed: st.ok() Non unique node name detected: SecondStageFeatureExtractor/InceptionV2/Mixed_5c/Branch_2/Conv2d_0c_3x3/weights

Answer 1

您可以尝试最新的转换器版本 1.3.1.1 我相信问题已得到解决。 https://github.com/tensorflow/tfjs/issues/2055

Answer 2

该转换器似乎仅适用于 Google Colab，没有任何问题。感谢大家的投入。