tensorflow
sjtu_leexx
这个作者很懒,什么都没留下…
展开
-
tf.nn.softmax_cross_entropy_with_logits()及tf.nn.sparse_softmax_cross_entropy_with_logits()函数的区别
这两个函数主要用于定义多分类问题的loss。tf.nn.softmax_cross_entropy_with_logits(_sentinel=None, labels=None, logits=None, dim=-1, name=None)解释该函数最重要的两个参数: 1. labels: 训练数据的标记真值,每一行为一个概率分布(可以直接使用one-hot编码) 2. logit...原创 2018-09-03 10:33:00 · 617 阅读 · 0 评论 -
tensorflow加载模型并测试的方法
利用tensorflow搭建模型并保存时,保存模型的方法为saver = tf.train.Saver()saver.save(sess, model_path + model_name)这样会在model_path路径下得到3个名为model_name的文件和一个checkpoint文件,例如,model_name=alexnet201809101818,则会得到如下四个文件 ...原创 2018-09-12 16:15:33 · 62761 阅读 · 4 评论 -
利用tensorflow训练网络时随机取batch数据的方法
利用tensorflow训练数据时,要自己手动给网络“喂”数据,在使用自己的数据集时,有时要手动选取batch数据,可利用如下代码选取随机batchdef get_batch(image, label, batch_size, now_batch, total_batch): if now_batch < total_batch: image_batch = im...原创 2018-09-10 15:22:45 · 5539 阅读 · 1 评论 -
tensorflow训练时取batch数据的方法
上一篇介绍了如何取随机batch,这一篇介绍的方法是先随机打乱数据,再按顺序一个一个batch的取。def shuffle_set(train_image, train_label, test_image, test_label): train_row = range(len(train_label)) random.shuffle(train_row) train...原创 2018-09-10 15:45:03 · 11517 阅读 · 0 评论 -
用tensorflow搭建AlexNet
最近研究了一下如何用tensorflow搭建alexnet并进行测试,使用tensorboard查看训练过程,我没有使用tensorflow自带的minist数据集,而是通过加载keras的minist数据集得到numpy array类型的数据,再自己处理之后“喂”个网络的,使用了tqdm显示训练进度条。我把整个过程写了一个完整的代码供大家参考。#!/usr/bin/env python2#...原创 2018-09-10 18:56:22 · 2917 阅读 · 3 评论 -
tensorflow在函数中用tf.Print输出中间值的方法
tensorflow由于其基于静态图的模式,导致写代码的时候很难调试,除了用官方的调试工具外,最直接的方法就是把中间结果输出出来查看,然而,直接用print函数只能输出tensor变量的形状,而不是数值,想要输出tensor的具体数值需要用tf.Print函数。网上有很多关于这个函数使用方法的说明,这里简要介绍:Print( input_, data, message=N...原创 2018-11-27 18:40:44 · 36971 阅读 · 10 评论