tensorflow
濯君
求知,内敛,有趣,勤静
展开
-
TensorFlow训练模型时,自定义数据处理函数,方便快速加载数据训练模型
一:自定义数据处理函数#txt:txt文件里面存放图像名称def read_images(txt, batch_size): imagepaths, labels = list(), list() #分别存放图像路径和label images_names = [] with open('/data/' + txt, 'r') as r: images_names.extend(r.readlines()) for name in images_names原创 2020-11-12 09:45:24 · 320 阅读 · 1 评论 -
tensorflow 设置输出log级别
如果不设置TensorFlow的输出log级别,默认情况下会输出所有信息,但是这种情况下往往有很多信息是我们不关注的,就会造成干扰。为了方便控制输出的信息,下面提供两种方法。一:默认是0输出所有信息,1输出info信息,2输出warning信息,3输出error信息import osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # or any {'0', '1', '2'}import tensorflow as tf二:这个直接设置输出信息级别tf.l原创 2020-11-09 20:31:14 · 582 阅读 · 0 评论 -
连续使用多个tf ckpt模型时,清理中间过程使用的模型图
连续使用多个tf ckpt模型时,若不清理前一个使用的模型,会导致使用下一个模型时出现问题,使用下面这行代码能清理之前使用的模型,注意要在session关闭之后再使用。tf.reset_default_graph()reference...原创 2020-11-09 20:19:53 · 191 阅读 · 0 评论 -
获取TensorFlow ckpt模型里的变量名称以及对应的值
一:输出模型中所有变量及其对应的值from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_fileprint_tensors_in_checkpoint_file(model_path='1dcnn.ckpt', tensor_name='', all_tensors=True)二:分别输出模型中的所有变量名称和值from tensorflow.python import pywrap_原创 2020-11-09 10:54:40 · 303 阅读 · 0 评论 -
TensorFlow直接使用ckpt模型predict不用restore
# -*- coding: utf-8 -*-# from util import *import cv2import numpy as npimport tensorflow as tf# from tensorflow.python.framework import graph_utilimport osos.environ['CUDA_DEVICE_ORDER'] = 'P...原创 2020-03-07 09:16:59 · 1057 阅读 · 2 评论 -
TensorFlow将ckpt模型转为pb模型
# -*- coding: utf-8 -*-import tensorflow as tffrom tensorflow.python.framework import graph_utilimport osos.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'os.environ['CUDA_VISIBLE_DEVICES'] = '1'imp...原创 2020-02-28 10:25:23 · 1603 阅读 · 3 评论 -
tf.Variable与tf.get_variable的区别
尽量使用tf.get_variable,当你想共享这个变量时,这样能够更容易重构代码。tf.Variable所在的层次更低,更接近底层,有时候tf.get_variable不能使用时,tf.Variable仍然能够使用。tf.Variable__init__( initial_value=None, trainable=None, collections=None,...原创 2019-07-13 22:52:05 · 326 阅读 · 0 评论 -
TensorBoard的基本使用
一:展示如何把TensorFlow的计算图写入日志import tensorflow as tfwith tf.name_scope('input1'): input1 = tf.constant([1.0, 2.0, 3.0], name='input1')with tf.name_scope('input2'): input2 = tf.Variable(tf....原创 2018-09-07 10:25:21 · 337 阅读 · 0 评论 -
(tensorflow)使用循环神经网络模型预测正弦函数
一:代码# -*- coding: utf-8 -*-import numpy as npimport tensorflow as tfimport matplotlib as mplmpl.use('Agg') #设置只保存绘制图片,不以窗口形式显示from matplotlib import pyplot as pltfrom tensorflow.contrib imp...转载 2018-09-07 09:50:31 · 1120 阅读 · 1 评论 -
TensorFlow 使用tf.QueueRunner,tf.Coordinator协同管理多线程队列
# -*- coding: utf-8 -*-import numpy as npimport tensorflow as tfimport threadingimport timequeue = tf.FIFOQueue(100, 'float')enqueue_op = queue.enqueue([tf.random_normal([1])])#表示需要启动5个线程,每个线...转载 2018-09-05 19:28:30 · 218 阅读 · 0 评论 -
TensorFlow多线程例子
# -*- coding: utf-8 -*-import numpy as npimport tensorflow as tfimport threadingimport timedef MyLoop(coord, worker_id): #使用tf.train.Coordinator类提供的协同工具判断当前线程是否需要停止 while not coord.shoul...转载 2018-09-05 19:09:54 · 602 阅读 · 0 评论 -
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
tensorflow版本与cuda不兼容导致的,若没有安装cuda9.0则需先安装,若安装了多个版本的cuda出了这个错,则需要设置环境变量。export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD...原创 2018-08-13 15:07:36 · 3754 阅读 · 0 评论