TensorFlow学习笔记

基于半个学期的深度学习,我将用老师曾经给过的一个项目来完成一次我的TensorFlow的学习笔记

首先,对于TensorFlow的理解,TensorFlow是一种较为完善的深度学习框架,支持所有流行变成语言开发,可以使用其内置函数使模型的搭建更加紧凑,此外,TensorFlow可在多平台上工作,允许将模型部署到工业生产中,并易于使用。

项目名称为基本分类:对服装图像进行分类,本项目是用TensorFlow搭建一个神经网络模型,对运动给和衬衫等服装图像进行分类

 

首先是导入该用的库

 

然后从keras中导入数据集

 

 此时加载的数据会返回四个Numpy数组

train_images(数据)和train_labels(标签)数组是训练集,用于模型学习

 

 图像是28x28的的Numpy的数组之间。像素值介到0到9之间,这些标签所对应的标签为

 

 接下来需要预处理数据,通过检查训练集的第一个图像,会发现像素值处于0到255之间

 

 

将这些值缩小到0到1之间,所以,将这些值➗255

 

 

接下来构建模型

首先需要设置层

神经网络的基本组成部分是。层会从向其馈送的数据中提取表示形式。希望这些表示形式有助于解决手头上的问题。

大多数深度学习都包括将简单的层链接在一起

 

 

该网络的第一层 tf.keras.layers.Flatten 将图像格式从二维数组(28 x 28 像素)转换成一维数组(28 x 28 = 784 像素)。将该层视为图像中未堆叠的像素行并将其排列起来。该层没有要学习的参数,它只会重新格式化数据。

展平像素后,网络会包括两个 tf.keras.layers.Dense 层的序列。它们是密集连接或全连接神经层。第一个 Dense 层有 128 个节点(或神经元)。第二个(也是最后一个)层会返回一个长度为 10 的 logits 数组。每个节点都包含一个得分,用来表示当前图像属于 10 个类中的哪一类

 

接下编译模型

在对模型进行训练之前。还需对其进行一些设置:

损失函数: 用于测量模型在训练期间的准确率

优化器 :决定模型如何根据其看到的数据和自身的损失函数进行更新

指标 - 用于监控训练和测试步骤

接下来训练模型:

训练模型分为四个步骤

  1. 将训练数据馈送给模型。在本例中,训练数据位于 train_images 和 train_labels 数组中。
  2. 模型学习将图像和标签关联起来。
  3. 要求模型对测试集(在本例中为 test_images 数组)进行预测。
  4. 验证预测是否与 test_labels 数组中的标签相匹配  

 

 向模型传送数据

 

接下来评估准确率

 

 传入测试数据,测试标签,设置为二维

 

最后进行预测

在模型经过训练后,可以使用它对一些图像进行预测

 

 现在做一个预测

predictions[0]

 

 

预测结果是一个包含 10 个数字的数组。它们代表模型对 10 种不同服装中每种服装的“置信度”

 

最后验证预测结果

 

 


posted @ 2022-04-24 21:46  doublemiracle  阅读(78)  评论(0编辑  收藏  举报