废话少说直接上代码样例如下importtensorflowastfimportosfromtensorflow.python.toolsimportfreeze_graph#本来这个model本无需解释太多,但是这么多人不能耐下心来看,那么我简单的说一下吧#network是你们自己定义的模型结构而已#ps:#defnetwork(input):#returntf.layers.max_pooling2d(input,2,2)frommodelimportnetworkos.environ['CUDA_VISIBLE_DEVICES']='2'#设置GPUmodel_path="pathto/model.ckpt-...
继续阅读 >
分类:tensorflow
2020
09-27
09-27
tensorflow实现将ckpt转pb文件的方法
本博客实现将自己训练保存的ckpt模型转换为pb文件,该方法适用于任何ckpt模型,当然你需要确定ckpt模型输入/输出的节点名称。 使用tf.train.saver()保存模型时会产生多个文件,会把计算图的结构和图上参数取值分成了不同的文件存储。这种方法是在TensorFlow中是最常用的保存方式。 例如:下面的代码运行后,会在save目录下保存了四个文件:importtensorflowastf#...
继续阅读 >
2020
09-27
09-27
tensorflow模型文件(ckpt)转pb文件的方法(不知道输出节点名)

2020
09-27
09-27
tensorflow指定CPU与GPU运算的方法实现

2020
09-27
09-27
TensorFLow 数学运算的示例代码
一、Tensor之间的运算规则相同大小Tensor之间的任何算术运算都会将运算应用到元素级不同大小Tensor(要求dimension0必须相同)之间的运算叫做广播(broadcasting)Tensor与Scalar(0维tensor)间的算术运算会将那个标量值传播到各个元素Note:TensorFLow在进行数学运算时,一定要求各个Tensor数据类型一致二、常用操作符和基本数学函数大多数运算符都进行了重载操作,使我们可以快速使用(+-*/)等,但是有...
继续阅读 >
2020
09-27
09-27
jupyter修改文件名方式(TensorFlow)

2020
09-27
09-27
python用TensorFlow做图像识别的实现

2020
09-27
09-27
Tensorflow中的降维函数tf.reduce_*使用总结
在使用tensorflow时常常会使用到tf.reduce_*这类的函数,在此对一些常见的函数进行汇总1.tf.reduce_sumtf.reduce_sum(input_tensor,axis=None,keep_dims=False,name=None,reduction_indices=None)参数:input_tensor:要减少的张量。应该有数字类型。axis:要减小的尺寸。如果为None(默认),则缩小所有尺寸。必须在范围[-rank(input_tensor),rank(input_tensor))内。keep_dims:如果为true,则保留长...
继续阅读 >
2020
09-27
09-27
jupyter notebook tensorflow打印device信息实例
juypternotebook中直接使用log_device_placement=True打印不出来device信息#Createsagraph.withtf.device('/device:CPU:0'):a=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[2,3],name='a')b=tf.constant([1.0,2.0,3.0,4.0,5.0,6.0],shape=[3,2],name='b')c=tf.matmul(a,b)#Createsasessionwithlog_device_placementsettoTrue.sess=tf.Session(config=tf.ConfigProto(log_device_pla...
继续阅读 >
2020
09-27
09-27
Tensorflow tf.nn.atrous_conv2d如何实现空洞卷积的




2020
09-27
09-27
tensorflow中tf.reduce_mean函数的使用
tf.reduce_mean函数用于计算张量tensor沿着指定的数轴(tensor的某一维度)上的的平均值,主要用作降维或者计算tensor(图像)的平均值。reduce_mean(input_tensor,axis=None,keep_dims=False,name=None,reduction_indices=None)第一个参数input_tensor:输入的待降维的tensor;第二个参数axis:指定的轴,如果不指定,则计算所有元素的均值;第三个参数keep_dims:是否降维度,设置为True...
继续阅读 >
2020
09-27
09-27
TensorFlow打印输出tensor的值
在学习TensorFlow的过程中,我们需要知道某个tensor的值是什么,这个很重要,尤其是在debug的时候。也许你会说,这个很容易啊,直接print就可以了。其实不然,print只能打印输出shape的信息,而要打印输出tensor的值,需要借助classtf.Session,classtf.InteractiveSession。因为我们在建立graph的时候,只建立tensor的结构形状信息,并没有执行数据的操作。一classtf.Session 运行tensorflow操作的类,其对象封装了执行...
继续阅读 >
2020
09-27
09-27
tensorflow常用函数API介绍
摘要:本文介绍了tensorflow的常用函数。1、tensorflow常用函数TensorFlow将图形定义转换成分布式执行的操作,以充分利用可用的计算资源(如CPU或GPU。一般你不需要显式指定使用CPU还是GPU,TensorFlow能自动检测。如果检测到GPU,TensorFlow会尽可能地利用找到的第一个GPU来执行操作.并行计算能让代价大的算法计算加速执行,TensorFlow也在实现上对复杂操作进行了有效的改进。大部分核相关的操作都是设备相关的实现,...
继续阅读 >