tensorflow
Ai君臣
通俗说技术AI,一针见血。AI改变世界!立志做一款让人偷懒依赖的个人定制的管家。可信可赖可知心。有志之士互勉!wx公号一休哥
展开
-
tensorflow checkpoint模型文件和pb文件加载,这样最好,预测时间短,不会泄露内存
背景:tensorflow需要加载checkpoint文件模型或者pb格式模型。 问题:怎么加载保存的checkpoint并预测,加载pb格式文件加载时多次预测会内存泄漏。 解决方法 1、checkpoint格式加载与预测 # 加载 self.graph = tf.Graph() # 为每个类(实例)单独创建一个graph with self.graph.as_default(): self.check_point_file = tf.train.lat原创 2020-09-11 10:08:03 · 768 阅读 · 0 评论 -
转tensorflow模型pb格式
一、模型格式 Tensorflow的保存分为三种:1. checkpoint模式;2. pb模式;3. saved_model模式。 Tensorflow中模型的保存一般使用tf.train.Saver()定义的存储器对象来保存魔心那个,并得到如下的文件: checkpoint model.ckpt.data-00000-of-00001 model.ckpt.index model.ckpt.meta # "model.ckpt" 是文件名 #保存时候生成三个文件,data对应的权重,meta是原创 2020-09-02 11:14:20 · 1503 阅读 · 4 评论