运行下列脚本，可以打印出模型各个节点变量的名称：
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join('model.ckpt-131805')
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print 'tensor_name: ',key
checkpoint_path为自己的模型路径
以上这篇tensorflow查看ckpt各节点名称实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。
在caffe中，如果使用的是c++接口，均值文件默认为.binaryproto格式，而如果使用的是python接口，均值文件默认的是numpy的.npy格式，在工作中有时需要将两者进行互...
model.py： #!/usr/bin/python # -*- coding: utf-8 -*- import torch from torch import nn imp...
在c语言中，一个进程可以fork出一个子进程，并让这个子进程exec一个新的命令。在python中，我们通过标准库的subprocess包来fork一个子进程，并在子进程中运行一个新的程...
本文实例讲述了Python通过公共键对字典列表排序算法。分享给大家供大家参考，具体如下： 问题：想根据一个或多个字典中的值来对列表排序 解决方案：利用operator模块中的itemge...
本文实例讲述了python访问mysql数据库的实现方法。分享给大家供大家参考，具体如下： 首先安装与Python版本匹配的MySQLdb 示例一 import MySQLdb co...