【问题标题】:Convert frozen graph to tensorflow-js format将冻结图转换为 tensorflow-js 格式
【发布时间】:2018-09-22 17:22:37
【问题描述】:

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

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

【问题讨论】:

    标签: tensorflow keras tensorflow-serving


    【解决方案1】:

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

    【讨论】:

      猜你喜欢
      • 2023-03-18
      • 2019-11-22
      • 2023-01-31
      • 1970-01-01
      • 1970-01-01
      • 2021-01-01
      • 1970-01-01
      • 2019-05-27
      • 2022-01-19
      相关资源
      最近更新 更多