CNN卷积神经网络

CNN卷积神经网络

#导入库函数
import tensorflow.keras
from tensorflow.keras.datasets import mnist,fashion_mnist
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense,Dropout,Flatten
from tensorflow.keras.layers import Conv2D,MaxPooling2D
from tensorflow.keras import backend as K

#加载并处理数据
(x_train,y_train),(x_test,y_test) = fashion_mnist.load_data()
batch_size = 128
num_classes = 10
epochs = 12
img_rows,img_cols = 28,28
if K.image_data_format()=='channels_first':
  x_train = x_train.reshape(x_train.shape[0],1,img_rows,img_cols)
  x_test = x_test.reshape(x_test.shape[0],1,img_rows,img_cols)
  input_shape = (1,img_rows,img_cols)
else:
  x_train = x_train.reshape(x_train.shape[0],img_rows,img_cols,1)
  x_test = x_test.reshape(x_test.shape[0],img_rows,img_cols,1)
  input_shape = (img_rows,img_cols,1)
x_train = x_train.astype('float32')
x_test = x_test.astype('float32')
x_train = x_train/255
x_test = x_test/255
print('x_train shape:', x_train.shape)
print("Training samples: {}".format(x_train.shape[0]))
print("Test samples: {}".format(x_test.shape[0]))

y_train = tensorflow.keras.utils.to_categorical(y_train,num_classes)
y_test = tensorflow.keras.utils.to_categorical(y_test,num_classes)

#建立模型（卷积神经网络）
model = Sequential()
model.add(Conv2D(32,kernel_size=(3,3),
          activation='relu',
          input_shape=input_shape))
model.add(Conv2D(64,kernel_size=(3,3),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.5))
model.add(Dense(num_classes,activation='softmax'))
model.summary()

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

import tensorflow as tf
import time

start_time = time.time()

#拟合并测试
model.fit(x_train,y_train,
      batch_size=batch_size,
      epochs = epochs,
      verbose = 2,
      validation_data=(x_test,y_test))
score = model.evaluate(x_test,y_test,verbose=0)
print('Test loss: {}'.format(score[0]))
print('Test accuracy: {}'.format(score[1]))

#训练时间
elapsed_time = time.time() - start_time
print("Elapsed time: {}".format(string(elapsed_time)))