TensorFlow
小飞鱼_bigdata
大数据、算法
展开
-
TensorFlow制作、读取TFRecord格式数据集
TFRecord数据格式是TensorFlow官方推荐的数据格式,不仅规范化读写,而且提高了IO效率。1.制作TFRecord数据原始数据为下图所示,文件夹名为类别标号,文件夹中存放的是各个类的图片:制作TFRecord的代码为:import osimport tensorflow as tffrom PIL import Imageimport numpy as npdef creat...原创 2018-04-02 23:58:47 · 1746 阅读 · 2 评论 -
TensorFlow实现AlexNet并持久化训练结果
1.参数设置数据的读取采用上篇博客中的方法https://blog.csdn.net/zryowen123/article/details/79796387,将上篇的代码命名为TFRecord.py,在本篇中进行了引用。import tensorflow as tfimport TFRecord as tfr # 自定义模块,用于进行tfrecords相关处理from tensorflow.p...原创 2018-04-10 23:28:19 · 1186 阅读 · 0 评论 -
TensorFlow用训练好的CNN模型检测
已预先训练好了一个识别猫、狗的二分类CNN模型,持久化在了一个cat_vs_dog.pb的文件中。现需要用其进行图片检测。1.代码读取训练好的pb文件,用来进行图片的检测,代码为:import tensorflow as tfimport numpy as npfrom PIL import ImageFILEPATH = 'F:/PycharmProjects/deepLearning/...原创 2018-04-11 00:21:18 · 5208 阅读 · 8 评论 -
CentOS6.7安装tensorflow遇到的问题:`GLIBCXX_3.4.19' not found
1.问题描述在CentOS6.7系统上利用Anaconda安装tensorflow,安装完后试运行报如下错:2.原因分析我们所依赖的动态库版本太低,可以用如下命令查看:strings /usr/lib64/libstdc++.so.6 | grep GLIBC输出:GLIBCXX_3.4GLIBCXX_3.4.1GLIBCXX_3.4.2GLIBCXX_3.4.3GLIBCXX_3.4.4GLIB...原创 2018-04-13 17:39:34 · 1969 阅读 · 0 评论 -
CentOS6.7 + GTX1070Ti + CUDA 9.0 + cuDNN 7.0.5搭建深度学习环境
显卡为GTX1070Ti,这是2017年下半年新出的卡,性能接近GTX1080。一.安装显卡驱动1.下载驱动到官网下载:https://www.geforce.com/drivers选择显卡型号下载后是一个名为NVIDIA-Linux-x86_64-390.48.run的文件2.安装编译环境gcc、kernel-devel、kernel-headers如果系统已经安装过了就不用再装了,这时需要注意...原创 2018-04-23 19:48:33 · 2855 阅读 · 0 评论 -
卷积神经网络文本分类模型TextCNN及Tensorflow实现
卷积神经网络的核心思想是捕捉局部特征。对于文本来说,局部特征就是由若干单词组成的滑动窗口,类似于N-gram。卷积神经网络的优势在于能够自动对N-gram特征进行组合和筛选,获得不同抽象层次的语义信息,并且由于权值共享机制,训练速度也比较快。用于文本分类任务的卷积神经网络模型为TextCNN,具体原理见论文Convolutional Neural Networks for Sentence C...原创 2019-03-12 00:11:06 · 2824 阅读 · 0 评论