TensorFlow
嘿芝麻
弱监督目标检测/定位
展开
-
win10+cuda9.0+cuDNN 7.0+Tensorflow1.5(GPU)安装
2018年初,谷歌在 GitHub 上正式发布了 TensorFlow 的最新版本 1.5.0,并开源了其代码。支持 CUDA 9 和 cuDNN 7 被认为是本次更新的最重要部分。 写在前面:检查GPU是否支持CUDA 先确定下自己的显卡型号(不要告诉我你不知道怎么查看自己的显卡型号)。 可以从下面的网址查看自己的显卡是否在支持之列。 如果你的显卡比较新,到这里...原创 2018-02-26 11:19:42 · 34142 阅读 · 31 评论 -
ubuntu16.04LTS + CUDA 8.0 + cudnn 5.1 + TensorFlow 安装(GPU版本)
写在前面 笔者安装环境: 系统:Ubuntu 16.04LTS (英文安装) 显卡:GTX 1080 NVIDIA驱动版本:NVIDIA-Linux-x86_64-390.25 anaconda版本:Anaconda3-4.2.0-Linux-x86_64 cuda版本:cuda_8.0.44_linux cuddn版本:cudnn-8.0-l...原创 2018-03-06 21:05:23 · 3167 阅读 · 0 评论 -
TensorFlow学习笔记之疑问解答(持续更新)
1、tensorflow中一个中括号和两个中括号是什么意思? b = tf.constant([3,3]) c = tf.constant([[3,3]]) with tf.Session() as sess: print(b,c) print(sess.run(b)) print(sess.run(c)) 返回结果: Tensor("Const_7:0", ...原创 2018-02-22 12:01:10 · 810 阅读 · 0 评论 -
Tensorflow: 读取ckpt文件中的tensor
在使用pre-train model时候,我们需要restore variables from checkpoint files. 经常出现在checkpoint 中找不到”Tensor name not found”. 这时候需要查看一下ckpt中到底有哪些变量 import os from tensorflow.python import pywrap_tensorflow m...转载 2018-07-16 14:44:14 · 2070 阅读 · 0 评论 -
tensorflow框架.ckpt .pb模型节点tensor_name打印及ckpt模型转.pb模型
转换模型首先要知道的是从哪个节点输出,如果没有源代码是很难清楚节点信息。 获取ckpt模型的节点名称 import os from tensorflow.python import pywrap_tensorflow checkpoint_path = os.path.join('./ade20k', "model.ckpt-27150") reader = pywrap_tensorf...原创 2018-08-29 17:06:15 · 13324 阅读 · 34 评论