TensorFlow实战
想好当码农了嘛
这个作者很懒,什么都没留下…
展开
-
tf.data.Dataset读取TFRecord文件进行训练
前言之前跑了一个小的CNN,因为训练数据量也不大,自己的电脑显卡2g,直接将数据读到内存中再取batch来训练还能直接跑;后来数据集大了一些,再用这种方法就不行了,内存不足。没办法,找了一些TFRecord和Dataset的资料把程序改了改,这样就可以边取数据边训练,总算能跑了,以后再也不用怕训练数据多啦。数据我的训练数据是一些元素为float类型的矩阵,用numpy生成一些来代表吧:da...原创 2019-11-02 20:37:18 · 1241 阅读 · 0 评论 -
tensorflow加载训练好的模型测试数据
总感觉耐不下心来好好学这些东西。计算图:TensorFlow中的所有计算都会被转化成计算图上的节点。张量:TensorFlow中,所有的数据都通过张量的形式来表示,可以简单地理解成多维数组。会话:TensorFlow中使用会话来执行定义好的计算。使用训练好的模型,最主要就是要得到输入和输出的接口。方法一:首先是输入。在训练网络时要给输入赋予name,比如:inputs_ = tf.pl...原创 2019-10-11 11:24:12 · 998 阅读 · 0 评论 -
TensorFlow实现inception-v3迁移学习
TensorFlow实现inception-v3迁移学习前言本来是打算跑通《TensorFlow实战Google深度学习框架》第二版的程序,但是没有成功,这个程序的思想是把inception-v3除了全连接层的其它层固定住,改变全连接层,前向传播和正常训练inception-v3一样,但反向传播更新参数只更新全连接层,最后自己训练后保存的模型也是包含inception-v3(除去它原本的全连接...原创 2019-10-12 22:14:49 · 788 阅读 · 0 评论