深入图像分类:使用美国手语数据集训练定制化神经网络

news/2024/9/25 5:08:20/

引言

在前一篇博客中,我们探讨了如何使用MNIST数据集训练一个基础的神经网络来进行手写数字识别。在本文中,我们将更进一步,使用美国手语字母表(ASL)数据集来构建一个定制化的图像分类模型。通过这个过程,你将学习到如何准备数据、构建模型、以及对模型进行训练和评估。

美国手语数据集简介

美国手语(ASL)是一种复杂的视觉语言,它包含了26个字母的手语表示。在本练习中,我们将使用一个包含这些字母图像的数据集,该数据集类似于MNIST,但对我们的模型来说更具挑战性。

数据准备

与MNIST数据集不同,ASL数据集需要我们手动加载和准备数据。我们将使用Pandas库来读取CSV格式的数据集,并使用matplotlib来可视化数据。
在这里插入图片描述

python">import pandas as pd
import matplotlib.pyplot as plt# 读取训练和验证数据集
train_df = pd.read_csv("data/asl_data/sign_mnist_train.csv")
valid_df = pd.read_csv("data/asl_data/sign_mnist_valid.csv")# 提取标签和图像数据
y_train = train_df['label']
y_valid = valid_df['label']
x_train = train_df.drop('label', axis=1).values
x_valid = valid_df.drop('label', axis=1).values# 数据可视化
plt.figure(figsize=(10,10))
for i in range(9):plt.subplot(3, 3, i+1)image = x_train[i].reshape(28, 28)plt.title(y_train[i], fontdict={'fontsize': 20})plt.axis('off')plt.imshow(image, cmap='gray')
plt.show()

数据归一化和分类编码

在训练模型之前,我们需要对图像数据进行归一化,并将标签转换为分类编码的形式。

python">from tensorflow.keras.utils import to_categorical# 数据归一化
x_train = x_train / 255.0
x_valid = x_valid / 255.0# 标签分类编码
num_classes = 24
y_train = to_categorical(y_train, num_classes)
y_valid = to_categorical(y_valid, num_classes)

构建模型

我们将构建一个包含两个隐藏层的神经网络模型,使用ReLU作为激活函数,并在输出层使用softmax函数。

python">from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Densemodel = Sequential()
model.add(Dense(units=512, activation='relu', input_shape=(784,)))
model.add(Dense(units=512, activation='relu'))
model.add(Dense(units=num_classes, activation='softmax'))

在这里插入图片描述

编译和训练模型

模型编译后,我们将使用交叉熵作为损失函数,并跟踪准确率指标。

python">model.compile(loss='categorical_crossentropy', metrics=['accuracy'])# 训练模型
history = model.fit(x_train, y_train, epochs=20, verbose=1, validation_data=(x_valid, y_valid))

结果讨论

训练过程中,我们可能会观察到模型在训练集上的准确率较高,但在验证集上的表现不佳。这可能是过拟合的迹象,即模型对训练数据过度拟合,而未能很好地泛化到未见数据。
在这里插入图片描述

结语

通过本文,你学习了如何处理新的数据集,并构建了一个针对特定任务的神经网络模型。在接下来的教程中,我们将探索卷积神经网络,这是一种在图像分类任务中更为强大的模型。


http://www.ppmy.cn/news/1453935.html

相关文章

docker-compose启动mysql5.7报错

描述一下问题经过: 使用docker compose 部署mysql5.7 文件如下: 使用命名卷的情况下,匿名卷不存在该问题 services:mysql:restart: alwaysimage: mysql:5.7container_name: mysql-devports:- 3306:3306environment:- MYSQL_DATABASEdev- MYSQL_ROOT_PAS…

面试官:flex:1代表什么

theme: channing-cyan 理解flex属性 在flexbox布局中,flex属性是用来控制项目的伸缩性的。它是一个复合属性,包含了flex-grow、flex-shrink和flex-basis这三个子属性。 在 Flexbox 中,每个子元素都拥有一个 “flex” 属性,用于控…

Java Jackson-jr 库是干什么用的

Jackson-jr 是一个轻量级的Java JSON 处理库。这个库被设计用来替代 Jackson 的复杂性。对比 Jackson 的复杂 API,Jackson-jr 的启动速度更快,包大小更小。 虽然Jackson databind(如ObjectMapper)是通用数据绑定的良好选择&#…

C#知识|如何在WinForm窗体中实现分割线绘制?

哈喽,你好啊,我是雷工! 在上位机UI设计中经常会用到分割线,用来分割界面区域。 像在KingSCADA、杰控、昆仑通态、WinCC、组态王、力控、易控等组态软件中非常简单,有现成的划线操作,选中相关工具直接绘制即…

使用FPGA发送一个经过曼彻斯特编码的伪随机序列

介绍 这几天突然就不知道要使用FPGA实现什么样的功能了,然后就跑去学习数电了,学的也是晕晕的。正好之前写了一个使用FPGA发送伪随机序列的代码,然后因为需要使用曼彻斯特编码,所以又加了一个模块吧,使得最后输出的波形经过曼彻斯特编码。 曼彻斯特编码 首先,曼彻斯特编…

Windows查找JDK的安装路径

如果很久之前安装了JDK,或者在别人的电脑上,想要快速指导JDK 的安装路径,可以通过啥方式指导JDK的安装路径是在哪里呢? 一、确认是否安装了JDK 首先我们打开命令行,如果输入 java -version 如果显示这种,…

速盾:高防CDN-完美防御

如今,网络安全已成为企业和个人不可忽视的重要问题之一。随着网络技术的不断发展,网络攻击的手段也越来越复杂和隐蔽。因此,保护网络的安全性变得尤为重要。针对网络攻击的防护手段也在不断升级和完善。其中,高防CDN作为一种完美的…

机器学习的指标评价

之前在学校的小发明制作中,在终期答辩的时候,虽然整个项目的流程都答的很流畅。 在老师提问的过程中,当老师问我recall,precision,accuracy等指标是如何计算的,又能够表示模型的哪方面指标做得好。我听到这个问题的时候&#xff…