saved_model_cli的使用

使用saved_model_cli命令可对导出的pb模型查看模型输入输出和直接运行得到预测结果。

假设我们保存的模型为：

1597977225|saved_model.pb|variables      |variables.data-00000-of-00001|variables.index

1. 查看模型输入输出：saved_model_cli show --dir 1597977225 --all

结果为：

signature_def['serving_default']:The given SavedModel SignatureDef contains the following input(s):inputs['UserID'] tensor_info:dtype: DT_INT64shape: (-1, 1)name: UserID:0inputs['ItemID'] tensor_info:dtype: DT_INT64shape: (-1, 1)name: ItemID:0The given SavedModel SignatureDef contains the following output(s):outputs['output'] tensor_info:dtype: DT_FLOATshape: (-1, 1)name: scores:0Method name is: tensorflow/serving/predict

2. 直接运行得到预测结果：

saved_model_cli run --dir model/1597977225 --tag_set serve --signature_def serving_default --input_exprs "UserID=[[73930]];ItemID=[[2050]]"

结果为：[[0.71090525]]

saved_model_cli的使用相关推荐

成功解决The scripts freeze_graph.exe, saved_model_cli.exe, tensorboard.exe, tflite_convert.exe, toco.exe
成功解决The scripts freeze_graph.exe, saved_model_cli.exe, tensorboard.exe, tflite_convert.exe, toco.exe ...
TensorFlow 2.2.0-rc0，这次更新让人惊奇！
AI编辑:我是小将刚刚谷歌在TensorFlow 开发者峰会上发布了 TensorFlow 2.2 版,2.2版本有很多地方的更新,我觉得可能两点会让大家欣喜若狂: 1. 增加同步BN层同步BN层 ...
使用CNN分类签名和文本图像
https://www.toutiao.com/a6644821136204907022/ 2019-01-10 18:59:42 正如我们所知道的任何具有法律重要性的文件,无论是合同,托运还是其他形 ...
Tensorflow: 保存和复原模型(save and restore)
报错: is not valid checkpoint 解决: module_file = tf.train.latest_checkpoint(diag_obj.save_path) saver.r ...
在NVIDIA Jetson Xavier NX上把yolov4-deepsort的模型pb模型使用tensorflow-onnx和onnx-tensorrt工具最终转换为tensorrt模型
文章目录: 1 安装tensorflow-onnx环境和把tensorflow的pb模型转换为onnx模型 1.1 安装tensorflow2onnx环境 1.2 把tensorflow的pb模型转换 ...
运行pip list报错：Fatal error in launcher: Unable to create process using...python\Scripts\pip.exe list
文章目录更新今天重装了电脑,将python系统变量添加了,但是pip打不开 !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! 更新下pip就好了- - python -m pip i ...
使用tensorflow serving部署keras模型（tensorflow 2.0.0）
点击上方"AI搞事情"关注我们内容转载自知乎:https://zhuanlan.zhihu.com/p/96917543 Justin ho 〉 Tensorflow 2.0.0 ...
AI繁荣下的隐忧——Google Tensorflow安全风险剖析
作者:[ Tencent Blade Team ] Cradmin 我们身处一个巨变的时代,各种新技术层出不穷,人工智能作为一个诞生于上世纪50年代的概念,近两年出现井喷式发展,得到各行各业的追捧,这 ...
Tensorflow Summary: 查看Tensorflow Model pb格式模型的信息
参考文链:如何查看Tensorflow SavedModel格式模型的信息; 参考:summary方法--小酒窝的博客; 参考:tensorflow中输出参数的方法--详细; tensorflow生成 ...

saved_model_cli的使用

saved_model_cli的使用相关推荐

最新文章

热门文章