将 bodypix js 模型转换为 pb 然后将 pb 转换为 tflite

问题描述

我想将 bodypix js 模型转换为 pb 。然后将 pb 转换为 tflite。

step1：下模

· 从 https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/model-stride16.json 下载 model.json 文件。

· 下载相应的权重来自 https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard1of23.bin ... https://storage.googleapis.com/tfjs-models/savedmodel/bodypix/resnet50/float/group1-shard23of23.bin

步骤2。转换pb文件

运行脚本： tfjs_graph_converter /Users/hh/Desktop/model /Users/hh/Desktop/model/model.pb 然后我在文件夹中得到model.pb。然后将model.pb重命名为saved_model.pb

步骤 3。转换 tflite 文件

运行脚本： tflite_convert --output_file /Users/hh/Desktop/model/model.tflite --graph_def_file /Users/hh/Desktop/model/saved_model.pb --saved_model_dir /Users/hh/Desktop/model

然后得到错误提示： 运行时错误：在 SavedModel 中找不到与标签 {'serve'} 关联的 MetaGraphDef。要检查 SavedModel 中的可用标签集，请使用 SavedModel CLI：saved_model_cli 可用标签：[]。

哪一步会出错？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）