第十一章使用卷积神经网络

最新推荐文章于 2024-09-29 20:27:25 发布

✘涛_

最新推荐文章于 2024-09-29 20:27:25 发布

阅读量84

点赞数

文章标签： cnn tensorflow keras

本文链接：https://blog.csdn.net/weixin_64072730/article/details/125115248

版权

from tensorflow.keras.datasets import mnist
from tensorflow.python.keras.utils import np_utils
import numpy as np
import matplotlib.pyplot as plt
np.random.seed(10)

(x_trian,y_train),(x_test,y_test) = mnist.load_data()

x_train4D = x_trian.reshape(x_trian.shape[0],28,28,1).astype("float32")
x_test4D = x_test.reshape(x_test.shape[0],28,28,1).astype("float32")

x_train4D_normalize = x_train4D / 255
x_test4D_normalize = x_test4D / 255

y_trainOneHot = np_utils.to_categorical(y_train)
y_testOneHot = np_utils.to_categorical(y_test)

from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense,Dropout,Flatten,Conv2D,MaxPooling2D
model = Sequential()

model.add(Conv2D(filters=6,
kernel_size=(5,5),
padding="same",
input_shape=(28,28,1),
activation="relu"))
model.add(MaxPooling2D(pool_size=(2,2)))
model.add(Conv2D(filters=16,
kernel_size=(5,5),
padding="same",
activation="relu"))
model.add(MaxPooling2D(pool_size=(2,2)))
model.add(Dropout(0.25))

model.add(Flatten())
model.add(Dense(128,activation="relu"))
model.add(Dropout(0.4))

model.add(Dense(10,activation="softmax"))
print(model.summary())

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

trian_history = model.fit(x=x_train4D_normalize,
y=y_trainOneHot,
validation_split=0.2,
epochs=10,
batch_size=300,
verbose=2)

def show_train_history( train_history , train, validation):
plt.plot(train_history.history[train])
plt.plot(trian_history.history[validation])
plt.title("Train history")
plt.ylabel(train)
plt.xlabel("Epoch")
plt.legend(["trian","validation"],
loc="upper left")
plt.show()
show_train_history(trian_history,"acc","val_acc")

scores = model.evaluate(x_test4D_normalize,y_testOneHot)
print("accuracy=",scores[1])

prediction = model.predict_classes(x_test4D_normalize)
print(prediction[:10])