TensorFlow实现打印每一层的输出


Posted in Python onJanuary 21, 2020

在test.py中可以通过如下代码直接生成带weight的pb文件,也可以通过tf官方的freeze_graph.py将ckpt转为pb文件。

constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def,['net_loss/inference/encode/conv_output/conv_output'])
with tf.gfile.FastGFile('net_model.pb', mode='wb') as f:
  f.write(constant_graph.SerializeToString())

tf1.0中通过带weight的pb文件与get_tensor_by_name函数可以获取每一层的输出

import os
import os.path as ops
import argparse
import time
import math
 
import tensorflow as tf
import glob
import numpy as np
import matplotlib.pyplot as plt
import cv2
 
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
gragh_path = './model.pb'
image_path = './lvds1901.JPG'
inputtensorname = 'input_tensor:0'
tensorname = 'loss/inference/encode/resize_images/ResizeBilinear'
filepath='./net_output.txt'
HEIGHT=256
WIDTH=256
VGG_MEAN = [103.939, 116.779, 123.68]
 
with tf.Graph().as_default():
  graph_def = tf.GraphDef()
  with tf.gfile.GFile(gragh_path, 'rb') as fid:
    serialized_graph = fid.read()
    graph_def.ParseFromString(serialized_graph)
 
    tf.import_graph_def(graph_def, name='')
 
    image = cv2.imread(image_path)
    image = cv2.resize(image, (WIDTH, HEIGHT), interpolation=cv2.INTER_CUBIC)
    image_np = np.array(image)
    image_np = image_np - VGG_MEAN
    image_np_expanded = np.expand_dims(image_np, axis=0)
 
    with tf.Session() as sess:
      ops = tf.get_default_graph().get_operations()
      tensor_name = tensorname + ':0'
      tensor_dict = tf.get_default_graph().get_tensor_by_name(tensor_name)
      image_tensor = tf.get_default_graph().get_tensor_by_name(inputtensorname)
      output = sess.run(tensor_dict, feed_dict={image_tensor: image_np_expanded})
      
      ftxt = open(filepath,'w')
      transform = output.transpose(0, 3, 1, 2)
      transform = transform.flatten()
      weight_count = 0
      for i in transform:
        if weight_count % 10 == 0 and weight_count != 0:
          ftxt.write('\n')
        ftxt.write(str(i) + ',')
        weight_count += 1
      ftxt.close()

以上这篇TensorFlow实现打印每一层的输出就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python通过exifread模块获得图片exif信息的方法
Mar 16 Python
Python的净值数据接口调用示例分享
Mar 15 Python
python3.6使用pickle序列化class的方法
Oct 22 Python
Python3实现的简单工资管理系统示例
Mar 12 Python
解决django服务器重启端口被占用的问题
Jul 26 Python
Window10下python3.7 安装与卸载教程图解
Sep 30 Python
django框架auth模块用法实例详解
Dec 10 Python
python设置环境变量的作用整理
Feb 17 Python
django从后台返回html代码的实例
Mar 11 Python
Python3之乱码\xe6\x97\xa0\xe6\xb3\x95处理方式
May 11 Python
Python利用Faiss库实现ANN近邻搜索的方法详解
Aug 03 Python
Python 实现Mac 屏幕截图详解
Oct 05 Python
NumPy排序的实现
Jan 21 #Python
tensorflow实现在函数中用tf.Print输出中间值
Jan 21 #Python
Python实现随机生成任意数量车牌号
Jan 21 #Python
tensorflow模型继续训练 fineturn实例
Jan 21 #Python
tensorflow ckpt模型和pb模型获取节点名称,及ckpt转pb模型实例
Jan 21 #Python
tensorflow查看ckpt各节点名称实例
Jan 21 #Python
python同义词替换的实现(jieba分词)
Jan 21 #Python
You might like
php中sql注入漏洞示例 sql注入漏洞修复
2014/01/24 PHP
PHP中new static()与new self()的区别异同分析
2014/08/22 PHP
php将日期格式转换成xx天前的格式
2015/04/16 PHP
PHP实现过滤各种HTML标签
2015/05/17 PHP
浅谈PHP中的
2016/04/23 PHP
我也种棵OO树JXTree[js+css+xml]
2007/04/02 Javascript
深入聊聊Array的sort方法的使用技巧.详细点评protype.js中的sortBy方法
2007/04/12 Javascript
锋利的jQuery 第三章章节总结的例子
2010/03/23 Javascript
JS定时器实例
2013/04/17 Javascript
下拉列表select 由左边框移动到右边示例
2013/12/04 Javascript
ExtJS4利根据登录后不同的角色分配不同的树形菜单
2014/05/02 Javascript
分享20款美化网站的 jQuery Lightbox 灯箱插件
2014/10/10 Javascript
JavaScript学习笔记之定时器
2015/01/22 Javascript
jquery 插件实现瀑布流图片展示实例
2015/04/03 Javascript
自己动手写的jquery分页控件(非常简单实用)
2015/10/28 Javascript
js省市联动效果完整实例代码
2015/12/09 Javascript
JavaScript:Array类型全面解析
2016/05/19 Javascript
浅谈JavaScript中小数和大整数的精度丢失
2016/05/31 Javascript
PassWord输入框代码分享
2016/06/07 Javascript
jQuery soColorPacker 网页拾色器
2016/06/22 Javascript
Angular2搜索和重置按钮过场动画
2017/05/24 Javascript
bootstrap datepicker插件默认英文修改为中文
2017/07/28 Javascript
javascript基于牛顿迭代法实现求浮点数的平方根【递归原理】
2017/09/28 Javascript
微信小程序之选项卡的实现方法
2017/09/29 Javascript
vue-cli中vue本地实现跨域调试接口
2019/01/16 Javascript
详解在网页上通过JS实现文本的语音朗读
2019/03/28 Javascript
微信小程序如何通过用户授权获取手机号(getPhoneNumber)
2020/01/21 Javascript
JS通用方法触发点击事件代码实例
2020/02/17 Javascript
Python实现删除时保留特定文件夹和文件的示例
2018/04/27 Python
python中安装django模块的方法
2020/03/12 Python
numpy 矩阵形状调整:拉伸、变成一位数组的实例
2020/06/18 Python
教师评优事迹材料
2014/01/10 职场文书
宿舍违规检讨书
2014/01/12 职场文书
出国考察邀请函
2014/01/21 职场文书
便利店投资创业计划书
2014/02/08 职场文书
专家推荐信模板
2014/05/09 职场文书