分类：ckpt

2020
09-29

tensorflow从ckpt和从.pb文件读取变量的值方式

最近在学习tensorflow自带的量化工具的相关知识，其中遇到的一个问题是从tensorflow保存好的ckpt文件或者是保存后的.pb文件(这里的pb是把权重和模型保存在一起的pb文件)读取权重，查看量化后的权重是否变成整形。因此将自己解决这个问题记录下来，为了下一次遇到时，可以有所参考，也希望给有需要的同学一个可能的参考。(1)从保存的ckpt读取变量的值(以读取保存的第一个权重为例)fromtensorflow.pythonimportpywrap_tensorflow... 继续阅读 >

2020
09-29

tensorflow转换ckpt为savermodel模型的实现

ckpt转换成SavedModelconvert_ckpt_to_savermodel.pyimporttensorflowastfimportsystrained_checkpoint_prefix=sys.argv[1]export_dir=sys.argv[2]graph=tf.Graph()config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True)withtf.compat.v1.Session(graph=graph,config=config)assess:#Restorefromcheckpointloader=tf.compat.v1.train.import_meta_graph(trained_checkpoint_prefi... 继续阅读 >

2020
09-27

tensorflow实现将ckpt转pb文件的方法

   本博客实现将自己训练保存的ckpt模型转换为pb文件，该方法适用于任何ckpt模型，当然你需要确定ckpt模型输入/输出的节点名称。   使用tf.train.saver()保存模型时会产生多个文件，会把计算图的结构和图上参数取值分成了不同的文件存储。这种方法是在TensorFlow中是最常用的保存方式。    例如：下面的代码运行后，会在save目录下保存了四个文件：importtensorflowastf#... 继续阅读 >