将冻结图转换为 tensorflow-js 格式
Convert frozen graph to tensorflow-js format
我有一个使用 Google 对象检测 API 的 SSD 模型（在自定义数据集上训练）。我冻结了一个生成几个文件（包括一个 *.pb 文件）的检查点。
问题：如何将冻结的推理图转换为 tf-js 可以使用的网络格式？
（PS ：官方网站确实在类似的行中提到了 example 但它期望 saved models 格式，而不是 frozen graph）
我找到了答案。这是一个两步转换过程 (1) 将检查点冻结为冻结图 input_type 为 encoded_image_string_tensor (help)。 (2) 现在，我们可以使用tensorflow JS exporter。
（注意：有可能step2会失败，因为不支持所有层都转换。）
我有一个使用 Google 对象检测 API 的 SSD 模型（在自定义数据集上训练）。我冻结了一个生成几个文件（包括一个 *.pb 文件）的检查点。
问题：如何将冻结的推理图转换为 tf-js 可以使用的网络格式？
（PS ：官方网站确实在类似的行中提到了 example 但它期望 saved models 格式，而不是 frozen graph）
我找到了答案。这是一个两步转换过程 (1) 将检查点冻结为冻结图 input_type 为 encoded_image_string_tensor (help)。 (2) 现在，我们可以使用tensorflow JS exporter。
（注意：有可能step2会失败，因为不支持所有层都转换。）