tensorflow
Murphy_0606
这个作者很懒,什么都没留下…
展开
-
tensorflow从.meta,.data,.index格式的模型文件中restore模型
例如模型保存在checkpoint文件夹下,保存的文件名分别为:model.index,model.data,model.meta with tf.Session(config=config) as sess: saver = tf.train.import_meta_graph('./checkpoint/model.meta')) save_pat...原创 2020-05-08 09:03:13 · 2517 阅读 · 3 评论 -
问题解决:writer.add_image添加的图片模糊,很多点点;原来是要uint8
保存图片:uint8类型转换训练过程中用writer.add_image向tensorboardX添加图片,发现图片总是下面这种鬼样子:但是在前一行用import matplotlib.pyplot as plt保存的图片就没有问题。经过查阅writer.add_image的源码,发现其要求输入的图片数据类型如下:The elements in img_tensor can either ...原创 2020-04-16 17:16:29 · 2886 阅读 · 0 评论 -
tensorflow保存模型时报错: ValueError: GraphDef cannot be larger than 2GB.
最终查出原因是因为输入网络的数据没有提前定义numpy的array接收,而是直接feed到模型里了,可能这样导致模型一次性保存的太大吧,在feed之前先用np.zeros()定义了数组,用来接收输入的图片数据,再传入network中,问题就能解决了。...原创 2019-10-28 09:20:48 · 699 阅读 · 1 评论 -
【已解决】中断打开tensorboard报错:ValueError: Duplicate plugins for name projector`
终端调用tensorboard时报错如下:raise ValueError('Duplicate plugins for name %s' % plugin.plugin_name)ValueError: Duplicate plugins for name projector`本以为是配置有问题,后来在网上找到了解决方案,经过尝试,验证了有效性,现在做一下记录。问题原因:这个问题主要是...原创 2019-09-12 14:55:55 · 14717 阅读 · 2 评论 -
包装用于TensorFlow操作的Python函数
最近使用tensoflow构建神经网络模型时遇到一个问题: 我们知道,tensorflow是一种计算图模型,即用图的形式来表示计算过程的一种模型。程序一般分为图的构建和图的执行两个阶段。图的构建阶段即图的定义阶段,会利用占位符tf.placeholder声明数据的格式和位置,用于传入数据到计算图中,执行阶段才会传入数据运行图模型。我在构建网络模型时需要用到非神经网络的函数,这个函数暂时在te...原创 2018-09-11 16:11:47 · 648 阅读 · 0 评论 -
pycharm运行tensorflow程序报错libcusolver.so.8.0: cannot open shared object file: No such file or directory
命令行下import tensorflow没问题,但是在pycharm中运行报错:libcusolver.so.8.0: cannot open shared object file: No such file or directory解决:在pycharm工程中run->edit configurations->Environment Variables,把cuda的路径添加到e...原创 2018-06-07 09:02:19 · 764 阅读 · 0 评论 -
tensorflow下怎么解决语义分割交叉熵损失总是nan
本次训练一个带有语义分割任务的网络,发现语义分割的交叉熵损失一直是nan,而且是从刚开始迭代就是nan,检查了网络一直没发现问题,学习率调小也不起作用,推测是损失函数计算中log传入了0,考虑到用的激活函数是tanh,会产生0值,所以考虑将激活函数换成不带0的sigmoid函数,换完之后发现立马就是正常损失值开始迭代了...原创 2018-05-06 18:37:07 · 1649 阅读 · 2 评论 -
Ubuntu14.04+tensorflow1.7+python3.6+CUDA8.0+CUDNN5.0(双版本tensorflow+双版本python)
前言自己电脑已有tensorflow1.2+python2.7+CUDA8.0+CUDNN5.0的实验环境,但新的实验要求tensorflow1.5以上+python3的环境,不想破坏已有的实验环境,所以想用anaconda环境配置一下新的,另外tensorflow1.5以上的已编译好的whl安装文件都要求CUDA9的版本,所以不想换CUDA版本的前提下选择源码编译tensorflow。综...原创 2018-04-19 10:48:04 · 1121 阅读 · 0 评论 -
tensorflow学习(一):Reduction以及tesorflow数学操作函数查找地址
tf.reduce_sumtf.reduce_sum(input_tensor, reduction_indices=None, keep_dims=False, name=None)Computes the sum of elements across dimensions of a tensor.# 'x' is [[1, 1, 1]]# [1, 1, 1]]tf.reduc原创 2017-10-11 19:38:58 · 418 阅读 · 0 评论 -
tensorflow:如何将tensor A 的形状reshape为tensor b 的形状
例如: tensor b的形状是(?, 128, 160, 6),想让tensor A的形状和b的一样。dim=[-1]+b.shape.as_list()[1:] #b的形状变成(-1,128,160,6)a=tf.reshape(a,shape=dim)原创 2017-09-07 22:17:40 · 1011 阅读 · 0 评论 -
tensorflow:flatten batch
tensorflow中的tensor的shape变换如将一个(?,128,160,6)的tensor转换(flatten)为(?,128×160×6)的shapeshape = x.get_shape().as_list() # a list: [?,128,160,6]dim = numpy.prod(shape[1:]) # dim = prod(128,1原创 2017-09-07 21:47:12 · 2783 阅读 · 0 评论 -
tensorflow学习笔记:ConfigProto&GPU(转自http://blog.csdn.net/u012436149/article/details/53837651)
欢迎使用Markdown编辑器写博客本Markdown编辑器使用StackEdit修改而来,用它写博客,将会带来全新的体验哦:Markdown和扩展Markdown简洁的语法代码块高亮图片链接和图片上传LaTex数学公式UML序列图和流程图离线写博客导入导出Markdown文件丰富的快捷键快捷键加粗 Ctrl + B 斜体 Ctrl + I 引用 Ctrl转载 2017-04-17 09:06:10 · 744 阅读 · 0 评论