当前位置:网站首页>深度学习21天——卷积神经网络(CNN):服装图像分类(第3天)
深度学习21天——卷积神经网络(CNN):服装图像分类(第3天)
2022-08-05 09:04:00 【清园暖歌】
目录
1.3 class_names[np.argmax(pre[1])]
一、新学习
1.1 卷积神经网络的图片输入形式
卷积神经网络(CNN)的输入是张量(Tensor)形式的(image_height,image_width,
color_channels),包含了图像高度、宽度及颜色信息。不需要输入batch size 。color_channels为(R,G,B)分别对应RGB的三个颜色通道(color channel)。在此示例中,我们的CNN输入,fashion_mnist数据集中的图片,形状是(28,28,1)即灰度图像。我们需要在声明第一层时将形状赋值给参数input_shape 。
model = models.Sequential([
layers.Conv2D(32, (3, 3), activation='relu', input_shape=(28, 28, 1)), #卷积层1,卷积核3*3
layers.MaxPooling2D((2, 2)), #池化层1,2*2采样
layers.Conv2D(64, (3, 3), activation='relu'), #卷积层2,卷积核3*3
layers.MaxPooling2D((2, 2)), #池化层2,2*2采样
layers.Conv2D(64, (3, 3), activation='relu'), #卷积层3,卷积核3*3
layers.Flatten(), #Flatten层,连接卷积层与全连接层
layers.Dense(64, activation='relu'), #全连接层,特征进一步提取
layers.Dense(10) #输出层,输出预期结果
])
model.summary() # 打印网络结构
所以在卷积层1,要把图片的shape值传入
1.2 关于编译
在准备对模型进行训练之前,还需要再对其进行一些设置。以下内容是在模型的编译步骤中添加的:
(1)损失函数(loss):用于测量模型在训练期间的准确率。您会希望最小化此函数,以便将模型“引导”到正确的方向上。
损失函数有预测值和实际的平方差(二元交叉熵)、平均平方差等
(2)优化器((optimizer)︰决定模型如何根据其看到的数据和自身的损失函数进行更新。
帮助实时更新参数
(3)指标(metrics):用于监控训练和测试步骤。以下示例使用了准确率,即被正确分类的图像的比率。
1.3 class_names[np.argmax(pre[1])]
如下所示
import numpy as np
a = np.array([3, 1, 2, 4, 6, 1])
b=np.argmax(a)#取出a中元素最大值所对应的索引,此时最大值位6,其对应的位置索引值为4,(索引值默认从0开始)
print(b)#4
参考:np.argmax()_wanghua609的博客-CSDN博客_np.argmax
所以 np.argmax(pre[1]) 即是取出测试集中第一个图像 对于10中服装的最大置信度 的索引值 i
再由 class_names[ i ]:取出名称
边栏推荐
猜你喜欢
随机推荐
How Entrepreneurs Attract Venture Capitalists
嵌入式实操----基于RT1170 移植memtester做SDRAM测试(二十五)
Rotation of the displayed value on the button
请问大佬们 ,使用 Flink SQL CDC 是不是做不到两个数据库的实时同步啊
sql server中 两表查询 平均数 分组
Example of Noise Calculation for Amplifier OPA855
基因数据平台
Weekly Report 2022-8-4
使用HBuilder离线本地打包ipa教程
Three solutions to solve cross-domain in egg framework
Detailed explanation of DNS query principle
这样写有问题吗?怎么在sql-client 是可以做到数据的同步的
Excuse me, guys, is it impossible to synchronize two databases in real time using Flink SQL CDC?
汇编语言(8)x86内联汇编
六年团队Leader实战秘诀|程序员最重要的八种软技能 - 脸皮薄容易耽误事 - 自我营销
Why do I recommend using smart async?
代码审计—PHP
tensorflow.keras无法引入layers
动态库之间回调函数使用
What is a good movie to watch on Qixi Festival?Crawl movie ratings and save to csv file