keras用CNN识别mnist

原创 2018年04月15日 21:57:18
import numpy as np
np.random.seed(1337) #重复性
from keras.datasets import mnist
from keras.utils import np_utils
from keras.models import Sequential
from keras.layers import Dense
from keras.layers import Activation,Convolution2D,MaxPool2D,Flatten
from keras.optimizers import Adam
from tensorflow.examples.tutorials.mnist import input_data

mnist = input_data.read_data_sets("/Users/wutao/PycharmProjects/手写数字识别/MNIST_data", one_hot=True)
X_train, Y_train = mnist.train.images, mnist.train.labels
X_test, Y_test = mnist.test.images, mnist.test.labels
X_train = X_train.reshape(-1, 1, 28, 28).astype('float32')
X_test = X_test.reshape(-1, 1, 28, 28).astype('float32')

# 搭建CNN
model = Sequential()

# conv1
model.add(Convolution2D(
    nb_filter=32,
    nb_row = 5,
    nb_col = 5,
    border_mode='same',
    input_shape= (1,28,28),
    ))
model.add(Activation('relu'))

# pooling1
model.add(MaxPool2D(
    pool_size=(2,2),
    strides=(3,3),
    border_mode='same'
    ))

# conv2
model.add(Convolution2D(64,5,5,border_mode="same"))
model.add(Activation('relu'))

# pooling2
model.add(MaxPool2D(pool_size=(2,2),border_mode='same'))

# 全连接层1
model.add(Flatten())
model.add(Dense(1024))
model.add(Activation('relu'))

# 全连接层2
model.add(Dense(10))
model.add(Activation('softmax'))

# optimizer
adam = Adam(lr=1e-4)

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

print('Training...................')
model.fit(X_train,Y_train,nb_epoch=1,batch_size=32)

print('\nTesting..................')
loss,accuracy = model.evaluate(X_test,Y_test)

print('test loss:',loss)
print('test accuracy:',accuracy)

利用keras(tensorflow) 做cnn mnist识别

keras图像数据处理以及图像识别小例子 1、数据预处理 数据集请自行下载,数据不大,20来兆 数据具体如下所示: 格式为 要识别的数字.序号.jpg 数据预处理代码,我用的是tens...
  • shizhengxin123
  • shizhengxin123
  • 2017-05-17 14:42:33
  • 4238

keras/构建卷积神经网络识别mnist

环境:Keras 2.04, python 2.7,GPU使用深度学习框架keras,构建卷积神经网络识别手写数字,keras在构建神经网络方面比Tensorflow简单很多,而且Tensorflow...
  • szj_huhu
  • szj_huhu
  • 2017-07-11 09:18:34
  • 676

Keras 深度学习框架Python Example:CNN/mnist

Keras是基于Theano的一个深度学习框架,它的设计参考了Torch,用Python语言编写,是一个高度模块化的神经网络库,支持GPU和CPU。...
  • Eric_Wilson
  • Eric_Wilson
  • 2015-11-15 15:39:19
  • 10278

keras实现简单CNN框架(mnist数据)

  • 2017年02月22日 22:53
  • 24.73MB
  • 下载

CNN实现手写数字库mnist

  • 2016年11月29日 14:27
  • 16KB
  • 下载

使用Keras搭建一个CNN处理MNIST数据

代码结构这里主要是参考了github上面一个u-net的程序的结构。链接 在项目中,程序员将整个神经网络分成了网络结构和训练两个类,并定义了一些函数来完成类似混淆矩阵生成这样的操作。 在这里,也是...
  • monsterhoho
  • monsterhoho
  • 2017-06-20 10:04:15
  • 1069

TensorFlow学习_02_CNN卷积神经网络_Mnist手写数字识别

github地址:https://github.com/lawlite19/MachineLearning_TensorFlow 或者上一篇Tensorflow的博客:http://blog.csdn...
  • u013082989
  • u013082989
  • 2016-12-16 16:54:40
  • 3936

卷积神经网络(CNN)的简单实现(MNIST)

卷积神经网络(CNN)的简单实现(MNIST)
  • fengbingchun
  • fengbingchun
  • 2016-03-06 19:20:43
  • 25660

基于深度学习框架Keras的CNN分类Mnist

在阅读了wepon博客的这篇博文 DeepLearning tutorial(6)易用的深度学习框架Keras简介 后,对keras的认识帮助甚大,但是由于keras快速的更新后,有些代码已经不适合...
  • lzhalan2016
  • lzhalan2016
  • 2016-08-27 11:04:14
  • 4585

03-Keras之用MNIST数据集训练一个CNN

03-Keras之用MNIST数据集训练一个CNN模型code# -*- coding: utf-8 -*-'''Trains a simple convnet on the MNIST datase...
  • sinat_25059791
  • sinat_25059791
  • 2017-02-19 18:31:44
  • 747
收藏助手
不良信息举报
您举报文章:keras用CNN识别mnist
举报原因:
原因补充:

(最多只允许输入30个字)