将冻结图转换为 tensorflow-js 格式

Posted 2023-03-17

【中文标题】将冻结图转换为 tensorflow-js 格式

【英文标题】：Convert frozen graph to tensorflow-js format

【发布时间】：2018-09-22 17:22:37

【问题描述】：

我有一个使用 Google 对象检测 API 的 SSD 模型（在自定义数据集上训练）。我冻结了一个生成几个文件（包括一个 *.pb 文件）的检查点。

问题：如何将冻结的推理图转换为 tf-js 可以使用的网络方便格式？ （PS：官方网站确实在类似的行中提到了example，但它需要saved models格式，而不是frozen graph）

【参考方案1】：

我找到了答案。这是一个两步转换过程 (1) 使用input_type 将检查点冻结为冻结图encoded_image_string_tensor (help)。 (2) 现在，我们可以使用 tensorflow JS 导出器了。 （注意：step2可能会失败，因为不是所有层都支持转换。）