当前位置:首页 >> 跨学科知识体系 >> 【VSCode部署模型】导出TensorFlow2.X训练好的模型信息,太平洋电脑论坛(vscode怎么导出项目)

【VSCode部署模型】导出TensorFlow2.X训练好的模型信息,太平洋电脑论坛(vscode怎么导出项目)

cpugpu芯片开发光刻机 跨学科知识体系 1
文件名:【VSCode部署模型】导出TensorFlow2.X训练好的模型信息,太平洋电脑论坛 【VSCode部署模型】导出TensorFlow2.X训练好的模型信息

参考tensorflow2.0 C++加载python训练保存的pb模型

经过模型训练及保存,我们得到“OptimalModelDataSet2”文件夹,模型的保存方法(.h5或.pb文件),参考【Visual Studio Code】c/c++部署tensorflow训练的模型

其中“OptimalModelDataSet2”文件夹保存着训练好的模型数据"saved_model.pb"

要在OptimalModelDataSet2所在的文件夹下执行:$ saved_model_cli show --dir ./OptimalModelDataSet2会得到:The given SavedModel contains the following tag-sets:'serve'使用'serve',继续执行:$ saved_model_cli show --dir ./OptimalModelDataSet2/ --tag_set serve会得到:The given SavedModel MetaGraphDef contains SignatureDefs with the following keys:SignatureDef key: "__saved_model_init_op"SignatureDef key: "serving_default"使用上述信息,继续:$ saved_model_cli show --dir ./OptimalModelDataSet2/ --tag_set serve --signature_def serving_default得到:The given SavedModel SignatureDef contains the following input(s):inputs['input_1'] tensor_info:dtype: DT_FLOATshape: (-1, 24)name: serving_default_input_1:0The given SavedModel SignatureDef contains the following output(s):outputs['dense_30'] tensor_info:dtype: DT_FLOATshape: (-1, 1)name: StatefulPartitionedCall:0Method name is: tensorflow/serving/predict这里输入和输出的name,要留好,c中调用会用到。

协助本站SEO优化一下,谢谢!
关键词不能为空
同类推荐
«    2025年12月    »
1234567
891011121314
15161718192021
22232425262728
293031
控制面板
您好,欢迎到访网站!
  查看权限
网站分类
搜索
最新留言
文章归档
网站收藏
友情链接