用CNN解决手写体数字识别。--tensorflow

本文介绍了使用卷积神经网络(CNN)解决手写数字识别问题的过程,对比了KNN算法的效果。通过Tensorflow构建CNN模型,包括数据处理、构建网络图、训练与验证。使用了卷积层、激活函数、池化层、全连接层和随机失活等技术,以提高模型的分类性能。实验中观察到过拟合现象,探讨了减少迭代次数和批量归一化的潜在改进策略。
摘要由CSDN通过智能技术生成

第一次解决这个问题是应用了机器学习中的Knn算法,发现排名并不是很高。再阅读了kaggle上其他人的一些笔记,知道了神经网络在图片分类上的优秀。我们可以用卷积层来得到数据图片中的某些特征,然后使用激活函数,再用汇聚层在空间维度上进行降维度操作。如此反复,最后使用全连接层,尺寸变为[1x1x10],就得到了10个类别的分类值。其中我们还是用了Adam来更新参数,用了随机失活防止过拟合。我使用了Tensorflow来搭建网络。
首先先引入一些需要的库。

import pandas as pd
import numpy as np

#加入%matplotlib inline才能在jupyter显示图像。
%matplotlib inline
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import tensorflow as tf
# settings。CS231n上介绍学习率初始化一般都为1e-4
LEARNING_RATE = 1e-4
# set to 20000 on local environment to get 0.99 accuracy
TRAINING_ITERATIONS = 2500        

DROPOUT = 0.5
BATCH_SIZE = 50

# set to 0 to train on all available data
VALIDATION_SIZE = 2000

# image number to output
IMAGE_TO_DISPLAY = 10

数据处理

kaggle上的数据并不是拿来就能用的,我们需要将它划分为像素,和标签。并且对像素进行归一化(不进行归一化可能会产生一些错误)。

# read training data from CSV file 
data = pd.read_csv('../input/train.csv')

print('data({0[0]},{0[1]})'.format(data.shape))
#展示前五行的数据
print (data.head())

这里写图片描述

#切片得到像素,不包括标签。
images = data.iloc[:, 1:].values
images = images.astype(np.float)

#归一化
images = np.multiply(images, 1.0 / 255.0)

print('images({0[0]},{0[1]})'.format(images.shape))

image_size = images.shape[1]

# in this case all images are square
image_width = image_height = np.ceil(np.sqrt(image_size)).astype(np.uint8)

>>>images(42000,784)

labels_flat = data.iloc[:,0].values.ravel()
#unique返回不同的数labels_count
labels_count = np.unique(labels_flat).shape[0]

print('labels_count => {0}'.format(labels_count))

>>>labels_count => 10
在分类问题中,我们经常使用one-hot vectors.在这个向量中,我们只有一个元素的值为1,其他的全为0。1的下标是什么就说明数字是多少。

#转化为one-hot vectors
#0 => [1 0 0 0 0 0 0 0 0 0 ]
#...
def dense_to_one_hot(labels_dense, num_classes):
    num_labels = labels_dense.shape[
TensorFlow是一个强大的开源机器学习库,常用于深度学习项目,包括手写数字识别手写数字识别是计算机视觉中的一个经典案例,它的目标是从图像中识别数字,如MNIST数据集中的手写字体。 在TensorFlow中实现手写数字识别通常包括以下步骤: 1. **数据预处理**:加载和处理MNIST数据集,将其分为训练集和测试集,通常是将图像标准化到0-1范围并调整尺寸。 2. **模型构建**:使用卷积神经网络(Convolutional Neural Network, CNN)作为基本架构,因为CNN对图像数据特别有效。可能包含多个卷积层、池化层、全连接层和激活函数(如ReLU)。 3. **编译模型**:设置损失函数(如交叉熵)、优化器(如Adam或SGD)和评估指标(如准确率)。 4. **训练模型**:使用训练集数据对模型进行迭代训练,调整权重以最小化损失。 5. **评估和验证**:在验证集上检查模型性能,防止过拟合。 6. **预测**:用训练好的模型对新的手写数字图片进行预测,并查看预测结果。 7. **可视化结果**:展示一些预测示例,对比真实标签和模型预测。 如果你对具体操作有兴趣,我可以提供一个简化的代码片段来展示如何使用TensorFlow进行手写数字识别: ```python import tensorflow as tf from tensorflow.keras.datasets import mnist # ... (数据预处理) # 创建一个简单的CNN模型 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.Flatten(), tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dropout(0.2), tf.keras.layers.Dense(10, activation='softmax') ]) # 编译模型 model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy']) # 训练模型 model.fit(train_images, train_labels, epochs=10, validation_data=(test_images, test_labels)) # 预测 predictions = model.predict(test_images) ```
评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值