我使用Google Object Detection API拥有SSD模型(使用自定义数据集进行培训)。我冻结了一个生成几个文件的检查点(包括一个* .pb文件)。
问题:如何将该冻结推理图转换为可由tf-js使用的web方便格式?
(PS:官方网站在类似的行上提到example,但它预期saved models
格式,而不是frozen graph
)
答案 0 :(得分:0)
我找到了答案。这是一个两步转换过程(1)将检查点冻结为冻结图,input_type
为encoded_image_string_tensor
(help)。 (2)现在,我们可以使用tensorflow JS导出器。
(注意:step2可能会失败,因为转换不支持所有层。)