基于Keras实现的卷积神经网络(CNN)示例
作者:微笑5885 | 来源:互联网 | 2024-12-03 19:35
本文介绍了一个使用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 更加高效且美观。我们将移除默认的广告和提示符,设置快捷键,并添加实用的别名和功能。 ...
[详细]
蜡笔小新 2024-11-28 07:25:46
本文详细介绍了如何在Apache Shiro框架中实现对用户登录失败重试次数的限制,通过自定义密码匹配器来增强系统的安全性。该方法不仅能够有效防止暴力破解攻击,还能确保合法用户的账户安全。 ...
[详细]
蜡笔小新 2024-12-03 21:39:23
本文介绍了如何处理在使用 aiohttp 进行 HTTPS 请求时遇到的 SSL 证书验证错误,包括忽略证书验证和使用自定义证书的方法。 ...
[详细]
蜡笔小新 2024-12-04 05:20:03
本文详细介绍了 C# 编程语言中 Main 方法的作用、不同形式及其使用场景,帮助开发者更好地理解和应用这一重要概念。 ...
[详细]
蜡笔小新 2024-12-03 00:07:55
一个产品数组拼图|集合 2 (O(1)空间) ...
[详细]
蜡笔小新 2024-12-02 19:18:16
本文详细介绍了Python中常用的图像处理库,包括scikit-image、Numpy、Scipy、Pillow、OpenCV-Python、SimpleCV、Mahotas、SimpleITK、pgmagick和Pycairo,旨在帮助开发者和研究人员选择合适的工具进行图像处理任务。 ...
[详细]
蜡笔小新 2024-12-02 17:15:21
本文介绍了如何通过十折交叉验证方法评估回归模型的性能。我们将使用PyTorch框架,详细展示数据处理、模型定义、训练及评估的完整流程。 ...
[详细]
蜡笔小新 2024-12-02 15:52:44
本文详细探讨了如何利用Python中的NumPy库来计算两个向量间的欧氏距离,并提供了具体的代码示例,旨在为开发者提供实用的技术指南。 ...
[详细]
蜡笔小新 2024-12-02 15:18:50
本文探讨了K近邻(KNN)算法中K值的选择对模型复杂度的影响,通过实验分析不同K值下的模型表现,旨在为KNN算法的应用提供指导。 ...
[详细]
蜡笔小新 2024-12-02 14:15:03
本文档详细介绍了如何在Flutter项目中集成和使用高德地图插件,包括安装、配置及基本使用方法。 ...
[详细]
蜡笔小新 2024-12-02 14:05:24
本文实例为大家分享了Python实现代码统计工具的具体代码,供大家参考,具体内容如下思路:首先获取所有文件,然后统计每个文 ...
[详细]
蜡笔小新 2024-12-01 21:22:48
本文详细介绍了 Go 语言的关键特性和编程理念,包括其强大的并发处理能力、简洁的语法设计以及高效的开发效率。 ...
[详细]
蜡笔小新 2024-11-28 18:57:18
本文详细探讨了DropBlock这一正则化方法在卷积神经网络中的应用与效果。通过结构化的dropout方式,即在特征图中连续区域内的单元同时被丢弃,DropBlock有效解决了传统dropout在卷积层应用时效果不佳的问题。更多理论分析及其实现细节可参考原文链接。 ...
[详细]
蜡笔小新 2024-11-28 11:54:39
本文档提供了使用ECharts库创建柱状图、饼图和双折线图的JavaScript函数。每个函数都详细列出了参数说明,并通过示例展示了如何调用这些函数以生成不同类型的图表。 ...
[详细]
蜡笔小新 2024-11-27 20:24:40
本文详细介绍了Python中的流程控制与条件判断技术,包括数据导入、数据变换、统计描述、假设检验、可视化以及自定义函数的创建等方面的内容。 ...
[详细]
蜡笔小新 2024-11-27 20:04:59