基于卷积神经网络的图像识别-案例实施1

案例描述

学习如何搭建CNN卷积神经网络,训练cifar-10数据,识别图片中的内容。

案例分析

cifar-10是由Hinton的学生Alex Krizhevsky和Ilya Sutskever整理的一个用于识别普适物体的小型数据集。一共包含 10个类别的 RGB 彩色图 片:飞机( airplane )、汽车( automobile )、鸟类( bird )、猫( cat )、鹿( deer )、狗( dog )、蛙类( frog )、马( horse )、船( ship )和卡车( truck )。图片由32×32彩色图像组成,每个类有6000个图像。cifar-10数据集中共有50000个训练图像和10000个测试图像。

案例实施

**1.**加载数据

from tensorflow.keras import datasets, layers, models

(train_images, train_labels), (test_images, test_labels) = datasets.cifar10.load_data()

**2.**搭建卷积神经网络

Keras是一个由Python编写的开源人工神经网络库,可以作为Tensorflow、Microsoft-CNTK和Theano的高阶应用程序接口,进行深度学习模型的设计、调试、评估、应用和可视化。Keras同样提供了序列化方法,可以方便的帮助搭建神经网络。

model = models.Sequential()

model.add(layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)))

model.add(layers.MaxPooling2D((2, 2)))

model.add(layers.Flatten())

model.add(layers.Dense(64, activation='relu'))

model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])

**3.**可视化训练过程

为了方便查看训练过程中准确率的变化,可以借助matplotlib来可视化训练过程。

相关推荐
摘星编程7 分钟前
CANN内存管理机制:从分配策略到性能优化
人工智能·华为·性能优化
likerhood14 分钟前
3. pytorch中数据集加载和处理
人工智能·pytorch·python
Robot侠15 分钟前
ROS1从入门到精通 10:URDF机器人建模(从零构建机器人模型)
人工智能·机器人·ros·机器人操作系统·urdf机器人建模
haiyu_y16 分钟前
Day 46 TensorBoard 使用介绍
人工智能·深度学习·神经网络
阿里云大数据AI技术21 分钟前
DataWorks 又又又升级了,这次我们通过 Arrow 列存格式让数据同步速度提升10倍!
大数据·人工智能
做科研的周师兄22 分钟前
中国土壤有机质数据集
人工智能·算法·机器学习·分类·数据挖掘
IT一氪23 分钟前
一款 AI 驱动的 Word 文档翻译工具
人工智能·word
lovingsoft26 分钟前
Vibe coding 氛围编程
人工智能
百***074532 分钟前
GPT-Image-1.5 极速接入全流程及关键要点
人工智能·gpt·计算机视觉
yiersansiwu123d1 小时前
AI二创的版权迷局与健康生态构建之道
人工智能