Tensorflow
文章平均质量分 71
深度学习框架Tensorflow使用相关文章
lishanlu136
关注机器视觉,深度学习图像分类,人脸识别,目标检测/跟踪等方面,我的github账号:https://github.com/lishanlu136,欢迎交流。
展开
-
查看.pb文件的结构
本文介绍将训练好的model.pb文件在tensorboard中展示其网络结构1. 从pb文件中恢复计算图import tensorflow as tfmodel = 'model.pb' #请将这里的model.pb文件路径改为自己的graph = tf.get_default_graph()graph_def = graph.as_graph_def()graph...原创 2018-09-11 15:33:10 · 10821 阅读 · 0 评论 -
Ubuntu16.04LTS安装TensorFlow(CPU与GPU联合版)
写在前面的话接触深度学习也有两三年时间了,开始在学校用Caffe,现在出来工作主要用TensorFlow,配置TensorFlow也有很多次了,Windows下,Ubuntu下都有,现把在Ubuntu下配置TensorFlow的步骤总结一下,虽然很简单~~第一步:查询计算机是否有nvidia显卡lspci | grep -i vga第二步:安装cuda和cudnn(在上一步中...原创 2018-09-12 16:30:10 · 375 阅读 · 0 评论 -
一篇文章弄懂tensorflow的Variab和get_variable,name_scope和variable_scope的区别
文章目录Variable与get_variablename_scope与variable_scopeVariable与get_variable直接上代码:#!/usr/bin/env python# -*- coding: utf-8 -*-"""@Time : 2018/11/13 9:18@Author : Li Shanlu@File : Va...原创 2018-11-15 09:26:10 · 901 阅读 · 0 评论 -
利用tensorflow在tensorboard中可视化图结构
在tensorboard中可视化图结构可视化pb模型的图import tensorflow as tfimport ospb_model = "your_pb_model_path"graph = tf.get_default_graph()graph_def = graph.as_graph_def()graph_def.ParseFromString(tf.gfile.FastG...原创 2019-03-25 19:14:08 · 1074 阅读 · 0 评论 -
tensorflow的freeze graph及inference graph_transforms
tensorflow模型优化一般利用tensorflow训练出来的模型包含4个文件:check_point文件,指定模型存放路径的文件.data文件,存放模型的权重文件.index文件,存放网络节点的索引.meta文件,存放网络图结构当我们在自己的电脑上实验时,固然可以先加载meta文件获得网络图结构,再加载.data文件加载权重,然后进行推理(inference)。但是在生产环境...原创 2019-03-26 19:36:26 · 2053 阅读 · 0 评论 -
tensorflow的freeze_graph方法
前言:一般利用tensorflow训练出来的结果都是权重和graph分开保存的,在发布产品的时候,我们需要把权重和graph固化在一起,这里就会用到freeze_graph啦。freeze_graph目前主要有两种方式:一种是通过bazel编译tensorflow/python/tools/freeze_graph.py文件,然后利用命令的方式把tf.train.write_graph()生...原创 2019-03-27 10:58:54 · 2158 阅读 · 3 评论 -
tensorflow实现quantization-aware training(伪量化,fake quantization)
前面一篇文章讲模型优化的时候有讲到量化模型,但那只是量化权重,在实际计算的时候还是会反量化回去,用float32位计算,没有进行实际意义上的定点运算。今天讲的这个方式是可以部署在移动端进行定点运算的,乘现在网上关于这方面资料很少,赶紧写一篇,求赞呀~~~源代码位置:tensorflow/contrib/quantize/github参考:https://github.com/tensorf...原创 2019-03-28 16:37:10 · 13419 阅读 · 48 评论 -
tensorflow的tf.nn.l2_normalize()与tf.norm()
tensorflow的归一化在很多时候都会用到,tensorflow代码里面也有多个实现方式,下面主要以L2归一化来讲解。tf.nn.l2_normalize()函数原型:l2_normalize(x, dim, epsilon=1e-12, name=None)x为输入的向量; dim为指定按哪个维度求l2范化,dim取值为0或1,0代表列,1代表行; epsilon为l2...原创 2019-04-22 17:07:23 · 15225 阅读 · 0 评论 -
tensorflow的tf.train.Saver()函数使用小技巧
tensorflow的saver是很重要的,不光在保存模型文件的时候用到,在微调网络的过程中,加载预训练模型的时候也会用到;下面就一些实际工程中遇到的问题做一些讲解。Saver类def __init__(self, var_list=None, reshape=False, sharded=Fal...原创 2019-04-22 18:00:02 · 802 阅读 · 0 评论