草庐IT

tensorflow-slim

全部标签

python - 在 TensorFlow 中获取矩阵的对角线

有没有办法在TensorFlow中提取方阵的对角线?也就是说,对于这样的矩阵:[[0,1,2],[3,4,5],[6,7,8]]我想获取元素:[0,4,8]在numpy中,这通过np.diag非常简单:在TensorFlow中,有一个diagfunction,但它只是在对角线上的参数中指定的元素形成一个新矩阵,这不是我想要的。我可以想象这是如何通过跨步来完成的……但我在TensorFlow中看不到张量的跨步。 最佳答案 在tensorflow0.8中,可以使用tf.diag_part()提取对角线元素(参见documentation

python - 如何在 TensorFlow 中获取张量中元素的计数?

我想获取张量中元素的个数,例如,t=[1,2,0,0,0,0](t是一个张量)。我可以通过在Python中调用t.count(0)来获得零的数量4,但在TensorFlow中,我找不到任何函数来执行此操作。我怎样才能得到零的数量? 最佳答案 目前TensorFlow中没有内置的计数方法。但您可以使用现有工具以如下方式完成:deftf_count(t,val):elements_equal_to_value=tf.equal(t,val)as_ints=tf.cast(elements_equal_to_value,tf.int32)

python - 无法加载 native TensorFlow 运行时。 python 3.5.2

我正在尝试在我的电脑上安装Tensorflow,我安装了Python3.5.264位、适用于Windows10的cuda_8.0.61和cudnn-8.0-windows10-x64-v6.0我使用“native”pip安装GPU版本的Tensorflow,然后我打开IDLE并使用“importtensorflowastf”进行测试,我收到以下错误我该如何解决这个问题?提前谢谢你Traceback(mostrecentcalllast):File"C:\Users\Admin\AppData\Local\Programs\Python\Python35\lib\site-package

python - 用于 TensorFlow 的 SSIM/MS-SSIM

TensorFlow是否有SSIM或什至MS-SSIM实现?SSIM(结构相似性指数指标)是衡量图像质量或图像相似性的指标。它受到人类感知的启发,并且根据几篇论文,与l1/l2相比,它是一个更好的损失函数。例如,参见LossFunctionsforNeuralNetworksforImageProcessing.到目前为止,我找不到TensorFlow中的实现。在尝试通过从C++或Python代码(例如Github:VQMT/SSIM)移植它来自己完成之后,我陷入了诸如将高斯模糊应用于TensorFlow中的图像的方法。已经有人尝试自己实现了吗? 最佳答案

python - Tensorflow 相同的训练精度持续

我被困在Tensorflow上的CNN模型上。我的代码如下。图书馆#-*-coding:utf-8-*-importtensorflowastfimporttimeimportjsonimportnumpyasnpimportmatplotlib.pyplotaspltimportrandomimportmultiprocessingasmpimportglobimportos型号definference(images_placeholder,keep_prob):defweight_variable(shape):initial=tf.truncated_normal(shape,s

python - 在 Tensorflow 中混合前馈层和循环层?

有没有人能够在Tensorflow中混合前馈层和循环层?例如:输入->转换-​​>GRU->线性->输出我可以想象一个人可以用前馈层定义他自己的单元,然后可以使用MultiRNNCell函数堆叠没有状态的单元,比如:cell=tf.nn.rnn_cell.MultiRNNCell([conv_cell,GRU_cell,linear_cell])这会让生活变得更轻松...... 最佳答案 您不能只执行以下操作吗:rnnouts,_=rnn(grucell,inputs)linearout=[tf.matmul(rnnout,weig

python - 如何在 TensorFlow 中使用稀疏张量进行计算?

我想以批处理方式从TensorFlow中的DNC实现中实现这个公式。对于批量密集张量,它非常简单。#w[B,N],p[B,N],L[B,N,N],B=batch_sizedot_prod=tf.batch_matmul(tf.expand_dims(w,axis=2),tf.expand_dims(p,axis=1))one_prod=1-tf.expand_dims(w,1)-tf.expand_dims(w,2)L=one_prod*pre_L+dot_prod有没有办法用稀疏张量来实现这个?w、p和L稀疏但TensorFlow缺少稀疏批量矩阵相乘和稀疏索引。

python - Tensorflow - 如何为 tf.Estimator() CNN 使用 GPU 而不是 CPU

我认为它应该与withtf.device("/gpu:0")一起使用,但我应该把它放在哪里?我不认为它是:withtf.device("/gpu:0"):tf.app.run()那么我应该把它放在tf.app的main()函数中,还是放在我用于估算器的模型函数中?编辑:如果这有帮助,这是我的main()函数:defmain(unused_argv):"""Codetoloadtrainingfoldsdatapickleorgenerateoneifnotpresent"""#CreatetheEstimatormnist_classifier=tf.estimator.Estimat

python - TensorFlow 内部错误 : Unable to get element as bytes

我正在尝试使用TensorFlow对一些包含分类和数字数据混合的日志数据运行DNNClassifier。我已经创建了特征列来指定和存储/散列tensorflow的数据。当我运行代码时,我收到“无法将元素作为字节获取”内部错误。注意:我不想删除此article中所述的Nan值所以我使用此代码将它们转换为0train=train.fillna(0,axis=0)所以我不确定为什么我仍然收到此错误。如果我删除Nan,那么它会起作用,但我不想删除Nan,因为我觉得模型需要它们进行训练。defcreate_train_input_fn():returntf.estimator.inputs.pa

python - Tensorflow:如何找到 tf.data.Dataset API 对象的大小

我理解数据集API是一种迭代器,它不会将整个数据集加载到内存中,因此它无法找到数据集的大小。我说的是存储在文本文件或tfRecord文件中的大量数据的上下文。这些文件通常使用tf.data.TextLineDataset或类似的东西读取。使用tf.data.Dataset.from_tensor_slices可以轻松找到加载的数据集的大小。我询问数据集大小的原因如下:假设我的数据集大小是1000个元素。批量大小=50个元素。然后训练步骤/批处理(假设1个纪元)=20。在这20个步骤中,我想将我的学习率从0.1呈指数衰减到0.01作为tf.train.exponential_decay(