tensorflow
文章平均质量分 78
YatChiu
学生
展开
-
Tensorflow使用笔记(2): 如何构建TFRecords并进行Mini Batch训练
引言前段时间在做一门课程的期末大作业的时候,用到了TensorFlow,构建了含有两层卷积层的神经网络去做 交通标志的识别,一开始使用 24x24 的图像作为输入(把数据集的图像都resize为24x24)后来感觉应该设计大一点会可靠一点,那就想把输入的图像都改为 64x64 的大小,相应修改了网络一些参数后,run的时候发现出问题了,我还以为是代码没有改好,仔细看一下提示信息:run out o原创 2017-06-24 17:10:53 · 2058 阅读 · 0 评论 -
Tensorflow使用笔记(1) 怎么保存和使用训练好的参数和模型
在TensorFlow中遇到的问题1如何保存好训练的结果:最近在学习Tensorflow 构建CNN,训练需要耗费时间,把训练好的各个参数保存下来是最简便的,网上有很多教程 假设会话为sess,计算图为graph 网上看了很多资料,使用 saver=tf.train.Saver() # 不传入参数代表默认存入全部参数saver.saver(sess,'model,ckpt')这样来保存模型原创 2017-06-03 12:14:20 · 13237 阅读 · 0 评论