# coding: utf-8
import numpy as np
from keras.datasets import mnist
from keras.utils import np_utils
from keras.models import Sequential
from keras.layers import Dense, Activation, Convolution2D, MaxPooling2D, Flatten
from keras.optimizers import Adam
np.random.seed(1337)
# download the mnist
(X_train, Y_train), (X_test, Y_test) = mnist.load_data()
# data pre-processing
X_train = X_train.reshape(-1, 1, 28, 28)/255
X_test = X_test.reshape(-1, 1, 28, 28)/255
Y_train = np_utils.to_categorical(Y_train, num_classes=10)
Y_test = np_utils.to_categorical(Y_test, num_classes=10)
# build CNN
model = Sequential()
# conv layer 1 output shape(32, 28, 28)
model.add(Convolution2D(filters=32,
kernel_size=5,
strides=1,
padding='same',
batch_input_shape=(None, 1, 28, 28),
data_format='
[深度学习框架] Keras上使用CNN进行mnist分类
最新推荐文章于 2023-11-01 15:48:41 发布