import tensorflow as tf from tensorflow import keras import matplotlib.pyplot as plt %matplotlib inline import numpy as np # 下載數(shù)據(jù)集并劃分為目標(biāo)集和測試集 (train_image,train_lable),(test_image,test_label) = tf.keras.datasets.fashion_mnist.load_data() train_image.shape

# 在最后一個維度擴張，擴張成一個4維數(shù)據(jù) train_image = np.expand_dims(train_image,-1) train_image.shape # (None,hight,witch,chanal)1表示黑白3彩色

test_image = np.expand_dims(test_image,-1) # 建立模型 model = tf.keras.Sequential()# 順序模型 model.add(tf.keras.layers.Conv2D(32,(3,3),input_shape=train_image.shape[1:],activation="relu"))# 建立卷積層 #每層建立32個卷積核，卷積核大小（3*3） #輸入圖片的形狀如（60000，28，28，1除去第0位的圖片個數(shù)）就是28*28*1 # 激活函數(shù)relu model.add(tf.keras.layers.MaxPool2D())# 最大池化默認(rèn)2*2形狀 model.add(tf.keras.layers.Conv2D(64,(3,3),activation="relu"))# 再次添加卷積層2的n次方形式添加卷積核 model.add(tf.keras.layers.GlobalAveragePooling2D())# 全局平均值池化 model.add(tf.keras.layers.Dense(10,activation="softmax"))# 輸出 model.summary()

# 訓(xùn)練模型 model.compile(optimizer="adam",loss="sparse_categorical_crossentropy",metrics=["acc"]) history = model.fit(train_image,train_lable,epochs=30,validation_data=(test_image,test_label))

history.history.keys()

# 正確率（通過繪圖得train數(shù)據(jù)得分不高未達到擬合 test數(shù)據(jù)過擬合） plt.plot(history.epoch,history.history.get("acc"),label="acc") plt.plot(history.epoch,history.history.get("val_acc"),label="val_acc")

# 誤差 plt.plot(history.epoch,history.history.get("loss"),label="loss") plt.plot(history.epoch,history.history.get("val_loss"),label="val_loss")

優(yōu)化模型

# CNN優(yōu)化增加卷積層及抑制擬合：增大測試訓(xùn)練集隱藏單元數(shù)增大擬合，降低抑制數(shù)據(jù)擬合# 建立模型 model = tf.keras.Sequential() model.add(tf.keras.layers.Conv2D(64,(3,3),input_shape=train_image.shape[1:],activation="relu",padding="same")) model.add(tf.keras.layers.Conv2D(64,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.Conv2D(64,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.MaxPool2D()) model.add(tf.keras.layers.Dropout(0.5)) model.add(tf.keras.layers.Conv2D(128,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.Conv2D(128,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.MaxPool2D()) model.add(tf.keras.layers.Dropout(0.5)) model.add(tf.keras.layers.Conv2D(256,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.Conv2D(256,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.MaxPool2D()) model.add(tf.keras.layers.Dropout(0.5)) model.add(tf.keras.layers.Conv2D(512,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.Conv2D(512,(3,3),activation="relu",padding="same")) model.add(tf.keras.layers.Dropout(0.5)) model.add(tf.keras.layers.GlobalAveragePooling2D()) model.add(tf.keras.layers.Dense(256,activation="relu")) model.add(tf.keras.layers.Dense(10,activation="softmax"))

總結(jié)

以上是生活随笔為你收集整理的深度学习-Tensorflow2.2-卷积神经网络{3}-卷积神经网络CNN示例-12的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。