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

时间:2018-04-12 13:57:45

标签: tensorflow keras tensorflow-serving

我使用Google Object Detection API拥有SSD模型(使用自定义数据集进行培训)。我冻结了一个生成几个文件的检查点(包括一个* .pb文件)。

问题:如何将该冻结推理图转换为可由tf-js使用的web方便格式? (PS:官方网站在类似的行上提到example,但它预期saved models格式,而不是frozen graph

1 个答案:

答案 0 :(得分:0)

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