tensorflow
MagiChos
这个作者很懒,什么都没留下…
展开
-
tf.nn.embedding_lookup函数
tf.nn.embedding_lookup(params, ids, partition_strategy='mod', name=None, validate_indices=True, max_norm=None)简单的来讲,就是将一个数字序列ids转化成embedding序列表示。 假设params.shape=[v,h], ids.shape=[m], 那么该函数会返回一个shape=原创 2017-11-16 16:09:01 · 522 阅读 · 0 评论 -
tf.stack和tf.unstack
tf.stack和tf.unstack分别表示矩阵的合并和分解,下面用一个小示例演示用法import tensorflow as tfimport sysimport osimport numpy as npa = tf.constant([1 , 2 , 3])b = tf.constant([4 , 5 , 6])c = tf.stack([a , b] , axis=0)d = tf原创 2017-12-23 21:17:21 · 1983 阅读 · 0 评论 -
tensorflow 部分函数笔记
tf.identity(input, name=None) 返回一个一模一样的新tensor 示例:import tensorflow as tfa = tf.constant(1.0 , tf.float32)init = tf.global_variables_initializer()b = tf.identity(a)with tf.Session() as sess:原创 2017-12-25 00:37:11 · 218 阅读 · 0 评论 -
tensorflow显存设置
为避免训练时占用全部的gpu显存,只需设置如下代码import numpy as npimport tensorflow as tfimport osos.environ['CUDA_VISIBLE_DEVICES'] = '0'# 程序最多占用制定gpu的50%显存gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.原创 2017-12-28 10:16:36 · 408 阅读 · 0 评论 -
tensorflow object detection模块
本文只是介绍一下tensorflow object detection模块的安装与使用。 1、下载tensorflow 的models模块https://github.com/tensorflow/models.git2、编译Protobuf库,在object_detection同级目录下打开终端运行: protoc object_detection/protos/*.prot...原创 2018-01-31 22:12:19 · 2402 阅读 · 0 评论 -
保存和加载pb模型
将模型保存为pbimport tensorflow as tffrom tensorflow.python.framework import graph_utillogdir='output/'with tf.variable_scope('conv'): w=tf.get_variable('w',[2,2],tf.float32,initializer=tf.ran原创 2018-01-31 22:20:07 · 5236 阅读 · 2 评论 -
tensorflow 物体识别
tensorflow推出的object detection模块,为我们做物体检测提供了极大的便利,如果还没有安装该模块,可以参照 http://blog.csdn.net/hanshuobest/article/details/79222685 下面的示例演示如何用已训练的模型做物体检测import numpy as npimport osimport tensorflow as原创 2018-02-04 18:09:09 · 5871 阅读 · 0 评论 -
tensorflow模型变量
#coding:utf-8import tensorflow as tffrom tensorflow.contrib import slimweights = slim.variable('weights' , shape=(3 , 3) , initializer=tf.truncated_normal_initializer(stddev=0.1) , regularizer=sli...原创 2018-03-22 17:45:39 · 441 阅读 · 1 评论 -
tensorflow教程——tf.GraphKeys
2018年05月21日 22:16:02 阅读数:317更多 个人分类: TensorFlow GraphKeys tf.GraphKeys包含所有graph collection中的标准集合名,有点像Python里的build-in fuction。首先要了解graph collection是什么。graph collection 在官方教程——图和会话中,介绍什么是tf.Gra...转载 2018-09-06 14:19:05 · 502 阅读 · 0 评论