0

0

基于机器学习算法进行图像分类的方法

王林

王林

发布时间:2023-05-08 18:13:08

|

1727人浏览过

|

来源于51CTO.COM

转载

服装数据集

服装数据集和MNIST数据集很像,有需要的可以查看教程《MNIST数据集》,包含70000个灰度图,每个图片28 x 28像素。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

使用机器学习对图片进行分类

时装数据集

在这里将使用60000张图片进行训练,使用10000张图片进行评估,可以直接使用Keras进行加载。

fashion_mnist = tf.keras.datasets.fashion_mnist(train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()

所有的图片可分为10个种类:

class_names = ['T-shirt/top', 'Trouser', 'Pullover', 'Dress', 'Coat', 'Sandal', 'Shirt', 'Sneaker', 'Bag', 'Ankle boot']

数据预处理:

在将数据送入到神经网络训练之前,需要对数据进行预处理,查看一张训练的图片,像素值的分布范围为[0, 255]

使用机器学习对图片进行分类

鞋子

对数据进行归一化处理:

train_images = train_images / 255.0test_images = test_images / 255.0

为了验证数据的正确性,展示前25张图片以及图片的分类:

微厦在线学习云服务平台2.0.0
微厦在线学习云服务平台2.0.0

微厦在线学习云服务平台 2.0是一款基于B/S架构的在线教育系统,将“在线视频学习、在线试题练习、在线同步考试”紧密相联,打造“学、练、考”于一体的在线教育系统,能够利用电脑、手机、微信等多种设备进行学习,方便学员利用碎片化时间进行随时随地的学习。并带有“分享、分润、分销”的辅助功能,对于平台推广、课程销售起到非常有效的帮助。适用类别:个人、单位、老师、大学、教育机构适用行业:企业培训、教育机构、

下载

使用机器学习对图片进行分类

服装

建立模型

神经网络基础模块就是层(Layer),层会从传递的数据中提取特征,这些特征对问题的解决很有帮助。

很多深度学习都是由一系列简单的层串联而成,大部分的层比如Dense,在训练过程中有可学习的参数。

model = tf.keras.Sequential([tf.keras.layers.Flatten(input_shape=(28, 28)),tf.keras.layers.Dense(128, activation='relu'),tf.keras.layers.Dense(10)])

Flatten层将二维(28 x 28)的图片转化为一维的数组,这一层没有参数可以学习,仅仅只是格式化数据。

第一个Dense层有128个节点或者说神经元,第二个Dense层返回长度为10的数组,每个节点包含当前图片属于哪个分类的得分。

模型编译

模型需要进行三个设置:

  1. 损失函数 - 这个主要用于评估模型在训练过程中的准确性
  2. 优化器 - 模型如何更新
  3. 量度 - 用于监测训练和测试步骤
model.compile(optimizer='adam',loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=['accuracy'])

填入训练数据:

model.fit(train_images, train_labels, epochs=10)

评估模型的正确性:

test_loss, test_acc = model.evaluate(test_images,test_labels, verbose=2)print('Test accuracy:', test_acc)

显示结果:

Test accuracy: 0.8835999965667725

训练过程输出:

1875/1875 [==============================] - 1s 523us/step - loss: 0.2379 - accuracy: 0.9110

可以看到测试数据的正确性是要略低于训练数据的正确性的,这个训练和测试的差距叫做overfitting(过拟合),过拟合发生在机器学习模型对于没有处理过的数据表现更差。

预测

模型训练之后,你可以使用它来对一些图片进行预测,添加一个Softmax层将结果转换为置信度,它更容易被理解

predictions = probability_model.predict(test_images)print(predictions[0])

可以看到第0张测试图片属于每个分类的置信度:

[4.7003473e-07 2.8711662e-09 1.8403462e-08 3.7643213e-09 2.0236126e-08 8.2177273e-04 1.0194674e-06 9.5114678e-02 2.8414237e-07 9.0406173e-01]

第9个数据的置信度最高,通过打印图片的标签也是9,说明预测正确。

随机选择一些图片输出:

使用机器学习对图片进行分类

置信度分布

第13张图片81%的可能性是凉鞋,说明机器学习预测错误,它应该是跑鞋。

也可以对单张图片进行预测,虽然是单张图片,但是Keras仍然需要数组进行传递,将图片添加到集合中。

img = (np.expand_dims(img, 0))

进行预测:

predictions_single = probability_model.predict(img)

总结

以上就是建立神经网络的简单过程,分为数据处理、模型的训练、预测等几个步骤。

相关专题

更多
php源码安装教程大全
php源码安装教程大全

本专题整合了php源码安装教程,阅读专题下面的文章了解更多详细内容。

7

2025.12.31

php网站源码教程大全
php网站源码教程大全

本专题整合了php网站源码相关教程,阅读专题下面的文章了解更多详细内容。

4

2025.12.31

视频文件格式
视频文件格式

本专题整合了视频文件格式相关内容,阅读专题下面的文章了解更多详细内容。

7

2025.12.31

不受国内限制的浏览器大全
不受国内限制的浏览器大全

想找真正自由、无限制的上网体验?本合集精选2025年最开放、隐私强、访问无阻的浏览器App,涵盖Tor、Brave、Via、X浏览器、Mullvad等高自由度工具。支持自定义搜索引擎、广告拦截、隐身模式及全球网站无障碍访问,部分更具备防追踪、去谷歌化、双内核切换等高级功能。无论日常浏览、隐私保护还是突破地域限制,总有一款适合你!

7

2025.12.31

出现404解决方法大全
出现404解决方法大全

本专题整合了404错误解决方法大全,阅读专题下面的文章了解更多详细内容。

42

2025.12.31

html5怎么播放视频
html5怎么播放视频

想让网页流畅播放视频?本合集详解HTML5视频播放核心方法!涵盖<video>标签基础用法、多格式兼容(MP4/WebM/OGV)、自定义播放控件、响应式适配及常见浏览器兼容问题解决方案。无需插件,纯前端实现高清视频嵌入,助你快速打造现代化网页视频体验。

4

2025.12.31

关闭win10系统自动更新教程大全
关闭win10系统自动更新教程大全

本专题整合了关闭win10系统自动更新教程大全,阅读专题下面的文章了解更多详细内容。

3

2025.12.31

阻止电脑自动安装软件教程
阻止电脑自动安装软件教程

本专题整合了阻止电脑自动安装软件教程,阅读专题下面的文章了解更多详细教程。

3

2025.12.31

html5怎么使用
html5怎么使用

想快速上手HTML5开发?本合集为你整理最实用的HTML5使用指南!涵盖HTML5基础语法、主流框架(如Bootstrap、Vue、React)集成方法,以及无需安装、直接在线编辑运行的平台推荐(如CodePen、JSFiddle)。无论你是新手还是进阶开发者,都能轻松掌握HTML5网页制作、响应式布局与交互功能开发,零配置开启高效前端编程之旅!

2

2025.12.31

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
R 教程
R 教程

共45课时 | 4.3万人学习

Bootstrap 5教程
Bootstrap 5教程

共46课时 | 2.7万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号