tensorflow
kether_
这个作者很懒,什么都没留下…
展开
-
tensorflow恢复模型
由于恢复模型时,按网上的操作存在很多问题,所以自己总结了一下:问题:训练时,加载测试数据,测试很正常,但训练完,重新恢复模型进行测试时存在很大的偏差,就像随机的结果。这是因为恢复时一些图设置(具体什么原因也没正清楚,但找到了正确的恢复方法)一.ckpt模型文件的恢复1.保存模型ckptsaver = tf.train.Saver(max_to_keep=1)with tf.Session() as sess: sess.run( tf.global_variables_initialize原创 2020-09-14 01:15:54 · 321 阅读 · 0 评论 -
MobileNet V2 tensorflow复现
前言在网上搜索很了很多也没看到关于tensorflow版的MobileNet V2网络架构。于是自己花了一天时间来完成了一个简单版本的复现。理论就不说了,搜其它资料吧网络结构如下:先理一下思路。MobileNet V2中最重要的就是bottleneck块。我们先看看块内内容1)可以看到input进去后会经过一个1*1的普通卷积来扩张维度。扩张倍数即网络结构表中的t。(表中为6倍)。其实...原创 2020-04-19 21:53:58 · 1025 阅读 · 0 评论 -
centos7下安装tensorflow-gpu版
我的版本CUDA 8.0 +cuDAA6.0+python3.5.1+tensorflow-gpu1.4.0说说遇到的坑(折腾了一下午):1.查资料说是python3.6可以与其他软件兼容,于是我就安装了python3.6.6版本,结果后面测试import tensorflow时会报错ImportError: libcublas.so.8.0: cannot open shared obje...原创 2019-03-07 11:28:45 · 623 阅读 · 0 评论 -
tensorflow使用自己的数据训练网络(一)
在学习tensorflow时,需要训练自己的数据,则需要自己写代码读入数据、组装成batch。主要分为两个步骤我的数据为三维的图像数据,后缀为nii。一般的二维图像数据也差不多是这个处理步骤一:制作TFrecord文件。假如目前在本地磁盘中有两类数据,已经按8:2的比例分为训练集和测试集,目录结构:/train /AD ...原创 2019-05-07 15:11:26 · 1457 阅读 · 0 评论 -
tensorflow读取csv文件及搭建神经网络训练
1.读取csv文件import tensorflow as tf # 生成一个先入先出队列和一个QueueRunner,生成文件名队列 def read_batch(filenames,batchsize): #filenames = ['mer_test/test_set.csv'] filename_queue = tf.train.string_inpu...原创 2019-06-19 11:51:50 · 6271 阅读 · 5 评论 -
tensorflow 实现计算混淆矩阵和各类评估指标
一 tf.confusion_matrix(y,pred_y)以二分类为例,混淆矩阵为2*2的矩阵,如果我们真实标签为real=[0,1,1,0,1],预测标签为predict=[0,1,0,1,1]tf.confusion_matrix(real,predict)可以得到一个混淆矩阵如果为one-hot编码则需要使用tf.argmax(y,1) real=tf.argmax...原创 2019-07-04 12:40:02 · 6692 阅读 · 4 评论