tensorflow保存模型以及调用模型[待测]

[数据] 2024-05-15 圈点870

摘要:tensorflow保存模型以及调用模型[待测]

tensorflow训练好的模型怎么调用?


1、实验室环境下,直接saver和restore即可。

2、生产环境:

(1)部署在移动终端上的(例如ios、android),场景:图像识别等。用freeze_graph合成pb和ckpt文件,然后用optimize_for_inference、quantize_graph进行优化。


再用TensorFlowInferenceInterface调用(这个,不知道ios和android是否相同)。

(2)部署在服务端提供服务使用的,场景:推荐系统等。使用tensorflow serving进行模型服务化。

看save的手册即可


示例:


tensorflow保存训练好的模型的代码如下:

sess = tf.Session()

saver = tf.train.Saver()

model_path = "D:\sample\model.ckpt"

save_path = saver.save(sess, model_path)


使用模型代码时,代码如下:

saver = tf.train.Saver()

saver.restore(sess, "D:\sample\model.ckpt")

result = sess.run(y, feed_dict={x: data})

y即为输出的结果。


tensorflow  

感谢反馈,已提交成功,审核后即会显示