基于卷积神经网络的图像识别-案例实施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来可视化训练过程。

相关推荐
稳石氢能15 小时前
稳石氢能董事长贾力出席2025高工氢电年会,呼吁制氢产业生态建设获广泛赞同。
人工智能
2301_8002561115 小时前
8.2 空间查询基本组件 核心知识点总结
数据库·人工智能·算法
Aspect of twilight16 小时前
PyTorch DDP分布式训练Pytorch代码讲解
人工智能·pytorch·python
用户51914958484516 小时前
滥用ESC10:通过注册表配置不当实现权限提升的ADCS攻击分析
人工智能·aigc
黎茗Dawn16 小时前
DDPM-KL 散度与 L2 损失
人工智能·算法·机器学习
玖日大大16 小时前
融合浪潮:从 “国产替代” 到 “范式创新” 的必然跃迁
人工智能
tomeasure16 小时前
INTERNAL ASSERT FAILED at “/pytorch/c10/cuda/CUDACachingAllocator.cpp“:983
人工智能·pytorch·python·nvidia
AI营销快线16 小时前
AI营销下半场:B2B选型指南
大数据·人工智能
小马爱打代码16 小时前
Spring AI:文生图:调用通义万相 AI 大模型
java·人工智能·spring
过河卒_zh156676616 小时前
算法备案最新通知:26年1月批备案号发放名单已锁定,发放前的复审抽审已开始
人工智能·算法·aigc·算法备案