上传者: 38590541
|
上传时间: 2021-09-17 11:47:47
|
文件大小: 68KB
|
文件类型: PDF
网上关于tensorflow模型文件ckpt格式转pb文件的帖子很多,本人几乎尝试了所有方法,最后终于成功了,现总结如下。方法无外乎下面两种:
使用tensorflow.python.tools.freeze_graph.freeze_graph
使用graph_util.convert_variables_to_constants
1、tensorflow模型的文件解读
使用tensorflow训练好的模型会自动保存为四个文件,如下
checkpoint:记录近几次训练好的模型结果(名称)。
xxx.data-00000-of-00001: 模型的所有变量的值(weights, b