我想将 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
可用标签:[]。
哪一步会出错?