热门标签 | HotTags
当前位置:  开发笔记 > 编程语言 > 正文

基于Keras实现的卷积神经网络(CNN)示例

本文介绍了一个使用Keras框架构建的卷积神经网络(CNN)实例,主要利用了Keras提供的MNIST数据集以及相关的层,如Dense、Dropout、Activation等,构建了一个具有两层卷积和两层全连接层的CNN模型。
### 基于Keras的CNN实现

本文展示了一个使用Keras框架构建的卷积神经网络(CNN)实例。该模型主要用于处理图像分类任务,特别是手写数字识别。通过使用Keras内置的MNIST数据集,我们构建了一个包含两层卷积层和两层全连接层的简单CNN模型。经过12轮训练后,模型的准确率达到了99.25%,展示了CNN在图像识别领域的强大能力。

```python
import numpy as np
np.random.seed(1337)
from keras.datasets import mnist
from keras.models import Sequential
from keras.layers import Dense, Dropout, Activation, Flatten
from keras.layers import Conv2D, MaxPooling2D
from keras.utils import np_utils
from keras import backend as K

batch_size = 128
num_classes = 10
epochs = 12
img_rows, img_cols = 28, 28
filters = 32
pool_size = (2, 2)
kernel_size = (3, 3)

(x_train, y_train), (x_test, y_test) = mnist.load_data()

if K.image_data_format() == 'channels_first':
x_train = x_train.reshape(x_train.shape[0], 1, img_rows, img_cols)
x_test = x_test.reshape(x_test.shape[0], 1, img_rows, img_cols)
input_shape = (1, img_rows, img_cols)
else:
x_train = x_train.reshape(x_train.shape[0], img_rows, img_cols, 1)
x_test = x_test.reshape(x_test.shape[0], img_rows, img_cols, 1)
input_shape = (img_rows, img_cols, 1)

x_train = x_train.astype('float32')
x_test = x_test.astype('float32')
x_train /= 255
x_test /= 255

y_train = np_utils.to_categorical(y_train, num_classes)
y_test = np_utils.to_categorical(y_test, num_classes)

model = Sequential()
model.add(Conv2D(filters, kernel_size, padding='same', input_shape=input_shape))
model.add(Activation('relu'))
model.add(Conv2D(filters, kernel_size))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=pool_size))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(num_classes))
model.add(Activation('softmax'))

model.compile(loss='categorical_crossentropy', optimizer='adadelta', metrics=['accuracy'])

model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, verbose=1, validation_data=(x_test, y_test))
score = model.evaluate(x_test, y_test, verbose=0)
print('Test score:', score[0])
print('Test accuracy:', score[1])
```

### Keras关键组件详解

#### 1. Keras的Backend支持
Keras支持多种后端,包括Theano和TensorFlow。不同的后端在数据格式上有所差异,例如Theano模式下,图像数据的形状为(样本数, 通道数, 高度, 宽度),而TensorFlow模式下则为(样本数, 高度, 宽度, 通道数)。这些差异在代码中通过`K.image_data_format()`来自动适应。

#### 2. 二维卷积层(Conv2D)
`Conv2D`层用于对二维输入进行卷积操作。主要参数包括:
- `filters`: 卷积核的数量。
- `kernel_size`: 卷积核的大小。
- `strides`: 卷积核的步长。
- `padding`: 边界填充方式,常见的有`'valid'`和`'same'`。

#### 3. 二维池化层(MaxPooling2D)
`MaxPooling2D`层用于对输入进行最大池化操作,减少数据的维度。主要参数包括:
- `pool_size`: 池化窗口的大小。
- `strides`: 池化窗口的步长。
- `padding`: 边界填充方式。

#### 4. 激活层(Activation)
`Activation`层用于对前一层的输出应用激活函数,常见的激活函数有`relu`、`sigmoid`、`tanh`等。

#### 5. Dropout层
`Dropout`层用于在训练过程中随机丢弃一部分神经元,以防止过拟合。主要参数包括:
- `rate`: 丢弃神经元的比例。

#### 6. Flatten层
`Flatten`层用于将多维输入展平为一维,常用于从卷积层到全连接层的过渡。

#### 7. 全连接层(Dense)
`Dense`层是一个全连接层,用于处理特征的线性组合。主要参数包括:
- `units`: 输出单元的数量。
- `activation`: 激活函数。

#### 8. 编译模型(compile)
`compile`方法用于配置模型的学习过程,主要参数包括:
- `optimizer`: 优化器。
- `loss`: 损失函数。
- `metrics`: 评估指标。

#### 9. 训练模型(fit)
`fit`方法用于训练模型,主要参数包括:
- `x`: 输入数据。
- `y`: 标签数据。
- `batch_size`: 批次大小。
- `epochs`: 训练轮数。
- `validation_data`: 验证数据。

#### 10. 评估模型(evaluate)
`evaluate`方法用于评估模型在测试集上的表现,主要参数与`fit`方法类似。

更多详细信息请参考Keras官方文档:
- 中文版:http://keras-cn.readthedocs.io/en/latest/
- 英文版:https://keras.io/
推荐阅读
  • 本文将指导你如何通过自定义配置,使 Windows Terminal 中的 PowerShell 7 更加高效且美观。我们将移除默认的广告和提示符,设置快捷键,并添加实用的别名和功能。 ... [详细]
  • Shiro功能拓展:登录失败重试次数限制
    本文详细介绍了如何在Apache Shiro框架中实现对用户登录失败重试次数的限制,通过自定义密码匹配器来增强系统的安全性。该方法不仅能够有效防止暴力破解攻击,还能确保合法用户的账户安全。 ... [详细]
  • 本文介绍了如何处理在使用 aiohttp 进行 HTTPS 请求时遇到的 SSL 证书验证错误,包括忽略证书验证和使用自定义证书的方法。 ... [详细]
  • 本文详细介绍了 C# 编程语言中 Main 方法的作用、不同形式及其使用场景,帮助开发者更好地理解和应用这一重要概念。 ... [详细]
  • 一个产品数组拼图|集合 2 (O(1)空间) ... [详细]
  • Python图像处理库概览
    本文详细介绍了Python中常用的图像处理库,包括scikit-image、Numpy、Scipy、Pillow、OpenCV-Python、SimpleCV、Mahotas、SimpleITK、pgmagick和Pycairo,旨在帮助开发者和研究人员选择合适的工具进行图像处理任务。 ... [详细]
  • 本文介绍了如何通过十折交叉验证方法评估回归模型的性能。我们将使用PyTorch框架,详细展示数据处理、模型定义、训练及评估的完整流程。 ... [详细]
  • 使用Python和NumPy高效计算向量间欧氏距离的方法
    本文详细探讨了如何利用Python中的NumPy库来计算两个向量间的欧氏距离,并提供了具体的代码示例,旨在为开发者提供实用的技术指南。 ... [详细]
  • 本文探讨了K近邻(KNN)算法中K值的选择对模型复杂度的影响,通过实验分析不同K值下的模型表现,旨在为KNN算法的应用提供指导。 ... [详细]
  • Flutter 高德地图插件使用指南
    本文档详细介绍了如何在Flutter项目中集成和使用高德地图插件,包括安装、配置及基本使用方法。 ... [详细]
  • 本文实例为大家分享了Python实现代码统计工具的具体代码,供大家参考,具体内容如下思路:首先获取所有文件,然后统计每个文 ... [详细]
  • 本文详细介绍了 Go 语言的关键特性和编程理念,包括其强大的并发处理能力、简洁的语法设计以及高效的开发效率。 ... [详细]
  • DropBlock:一种卷积网络的正则化技术
    本文详细探讨了DropBlock这一正则化方法在卷积神经网络中的应用与效果。通过结构化的dropout方式,即在特征图中连续区域内的单元同时被丢弃,DropBlock有效解决了传统dropout在卷积层应用时效果不佳的问题。更多理论分析及其实现细节可参考原文链接。 ... [详细]
  • ECharts图表绘制函数集
    本文档提供了使用ECharts库创建柱状图、饼图和双折线图的JavaScript函数。每个函数都详细列出了参数说明,并通过示例展示了如何调用这些函数以生成不同类型的图表。 ... [详细]
  • 本文详细介绍了Python中的流程控制与条件判断技术,包括数据导入、数据变换、统计描述、假设检验、可视化以及自定义函数的创建等方面的内容。 ... [详细]
author-avatar
微笑5885
这个家伙很懒,什么也没留下!
PHP1.CN | 中国最专业的PHP中文社区 | DevBox开发工具箱 | json解析格式化 |PHP资讯 | PHP教程 | 数据库技术 | 服务器技术 | 前端开发技术 | PHP框架 | 开发工具 | 在线工具
Copyright © 1998 - 2020 PHP1.CN. All Rights Reserved | 京公网安备 11010802041100号 | 京ICP备19059560号-4 | PHP1.CN 第一PHP社区 版权所有