卷积神经网络(CNN)的训练及代码实现

本文部分内容来自zouxy09的博客,谢谢!http://blog.csdn.net/zouxy09/article/details/9993371

以及斯坦福大学深度学习教程:http://ufldl.stanford.edu/wiki/index.php/UFLDL教程

CNN结构的连接比权值多很多,因为权值共享。CNN通过数据驱动的方式学习得到一些滤波器,作为提取输入的特征的一种方法。

典型CNN中开始几层都是卷积和下采样交替,然后在最后是一些全连接层。在全连接层时已经将所有两维特征map转化为全连接一维输入。

1、前向传播

假设该网络能处理c类分类问题,共N个训练样本。定义平方误差代价函数:

2、反向传播


调整参数。

批量梯度下降法是一种常用的优化目标函数的方法,通过对目标函数关于参数求导,来更新参数,其目标函数延梯度下降的方向快速逼近最小值。所以每次迭代都按照如下公式对


反向传播算法的思路如下:



3、卷积神经网络训练参数时的不同处

3.1卷积层

CNN中卷积层的BP更新。在卷积层,上层的特征map被一个可以学习的卷积核进行卷积,然后通过一个激活函数,就可以得到输出特征map。每个输出map可以组合卷积多个输入map正向传播时如下计算:



  • 5
    点赞
  • 40
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是一个用于分类手写数字的CNN卷积神经网络的Python代码示例: ``` import numpy as np import tensorflow as tf from tensorflow.keras.datasets import mnist # 加载数据集 (x_train, y_train), (x_test, y_test) = mnist.load_data() # 数据预处理 x_train = x_train.reshape(-1, 28, 28, 1).astype('float32') / 255.0 x_test = x_test.reshape(-1, 28, 28, 1).astype('float32') / 255.0 y_train = tf.keras.utils.to_categorical(y_train) y_test = tf.keras.utils.to_categorical(y_test) # 构建模型 model = tf.keras.models.Sequential([ tf.keras.layers.Conv2D(32, (3,3), activation='relu', input_shape=(28,28,1)), tf.keras.layers.MaxPooling2D((2,2)), tf.keras.layers.Conv2D(64, (3,3), activation='relu'), tf.keras.layers.MaxPooling2D((2,2)), tf.keras.layers.Conv2D(64, (3,3), activation='relu'), tf.keras.layers.Flatten(), tf.keras.layers.Dense(64, activation='relu'), tf.keras.layers.Dense(10, activation='softmax') ]) # 编译模型 model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) # 训练模型 model.fit(x_train, y_train, epochs=5, batch_size=64, validation_data=(x_test, y_test)) # 评估模型 test_loss, test_acc = model.evaluate(x_test, y_test) print('Test accuracy:', test_acc) ``` 这个CNN卷积神经网络模型包含了三个卷积层和两个全连接层,用于对手写数字进行分类。首先加载MNIST数据集并进行预处理,然后构建模型并编译。训练模型时使用了5个epochs,batch size为64。最后评估模型的表现并输出测试准确率。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值