码迷,mamicode.com
首页 > 其他好文 > 详细

记录二:tensorflow2.0写MNIST手写体

时间:2019-09-29 21:36:51      阅读:87      评论:0      收藏:0      [点我收藏+]

标签:ima   flow   ras   loading   引入   oid   ini   book   tensor   

  最近学习神经网络,tensorflow,看了好多视频,查找了好多资料,感觉东西都没有融入自己的思维中。今天用tensorflow2.0写了一个MNIST手写体的版本,记录下学习的过程。

复现手写体识别的基本步骤:准备数据,处理数据,搭建模型,迭代训练模型,使用模型。

 

一、全连接模型  

1、导入数据集(此次编写运行均在notebook)

  先引入所需模块,在用tf.keras下载数据

  技术图片

  2、分析图片的大小

  1)用shape查看数量大小;2)通过指定[0][1]来查看具体图像尺寸,lable存储的是image对应的数值

  技术图片

 

 

 

  3、处理数据

  就将原先的0-9label转换为独热编码,独热编码即 One-Hot 编码,又称一位有效编码,其方法是使用N位状态寄存器来对N个状态进行编码,每个状态都由他独立的寄存器位,并且在任意时候,其中只有一位有效。

    技术图片技术图片

 

  4、构建模型

  1)tf.keras.Sequential()构建一个顺序模型(此处模型的层数参照3blueBrown4层)

  2)第一层输入层是28*28个神经元,由传入图像的像素大小决定,Flatten是用来格式化数据的,将数据转化为一列。

  3)第二层和第三层各16个神经元,激活函数定义为sigmoid

  4)第四层输出层0-9,一共输出10个不同的,softmax是概率分布型

  技术图片

 

 

 

 5、训练模型 

  1)model.compile(loss=‘目标函数‘, optimizer=optimizer, metrics=[‘accuracy‘])

  2)fit( x, y, batch_size=32, epochs=10, verbose=1, callbacks=None, validation_split=0.0, validation_data=None, shuffle=True, class_weight=None, sample_weight=None, initial_epoch=0)

  技术图片

  6、图观模型

  一百次后,测试集跟训练集的准确性图示

技术图片

二、卷积模型

卷积网络大体布置为卷积层-池化层-全连接层。基础知识储备如何进行卷积前的估算,并且对相关数据进行调整,添加一层channel,确保数据一致性。

       2.1基础知识储备(卷积参数参照LeNet-5)

           技术图片

 

 

 

       2.2加载并对数据进行处理,与全连接不同的是,对数据还要再添加一层channel,手写体是黑白图片,故而设置为1。

      技术图片

 

 

 

       2.3参照上面的卷积处理,建立模型,并运行模型

   技术图片

 

 

  2.4 模型运行结果图示

 

   技术图片

 

记录二:tensorflow2.0写MNIST手写体

标签:ima   flow   ras   loading   引入   oid   ini   book   tensor   

原文地址:https://www.cnblogs.com/skylarzhan/p/11565214.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!