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

时间:2021-03-10 03:32:18

标签: bodypix

我想将 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 可用标签:[]。

哪一步会出错?

0 个答案:

没有答案
相关问题