huanggua16的博客

第一种保存pb方法 写入序列化的 PB 文件： with tf.gfile.FastGFile(pb_file_path+'model.pb', mode='wb') as f: f.write(constant_graph.SerializeToString())第二种保存pb方法： ...

（1）将保存的模型文件解析为GraphDef ：graph_def.ParseFromString(gfile.FastGFile(“model.pb”,‘rb’).read()),这里分为两步，一：通过gfile.FastGFile(“model.pb”,‘rb’).read()获取保存的pb模型对象并读取文件，二： graph_def = tf.GraphDef()创建graph_def对...

title: ckpt模型转pb模型手册tags: tensorflow,ckpt,pb为什么要把ckpt格式转为pb格式？ckpt格式的模型图结构和权重参数模型是分开保存在不同的文件里，而pb模型的图结构和权重参数的模型是保存在同一个文件中的,很多时候我们需要一个既包含图又包含变量的模型文件ckpt格式模型包含的文件如下：我们在checkpoint_dir目录下保存的文件结构如下：...