标签：layers plt lenet5 labels train lenet images cv

实现

import tensorflow as tf
from tensorflow import keras

from tensorflow.keras import datasets,layers,models
import matplotlib.pyplot as plt

(train_images,train_labels),(test_images,test_labels) = datasets.cifar10.load_data()

#normalize
train_images,test_images=train_images/255.0,test_images/255.0

#verify Data
class_names=['airplane','automobile','bird','cat','deer','dog','frog','horse','ship','truck']

plt.figure(figsize=(10,10))
for i in range(25):
  plt.subplot(5,5,i+1)
  plt.xticks([])
  plt.yticks([])
  plt.grid(False)
  plt.imshow(train_images[i],cmap=plt.cm.binary)
  plt.xlabel(class_names[train_labels[i][0]])
plt.show

#变换label为1-hot编码
train_labels=keras.utils.to_categorical(train_labels,10)
test_labels=keras.utils.to_categorical(test_labels,10)

train_images[0].shape
train_labels.shape

lenet_input = keras.Input(shape=(32,32,3),name='img')
x=layers.Conv2D(6,5,activation='relu')(lenet_input)
x=layers.MaxPooling2D(2)(x)
x=layers.Conv2D(16,5,activation='relu')(x)
x=layers.MaxPooling2D(2)(x)
#x=layers.Conv2D(120,5,activation='relu')(x)
x=layers.Flatten()(x)
x=layers.Dense(120,activation='relu')(x)
x=layers.Dense(84,activation='relu')(x)
lenet_output=layers.Dense(10,activation='softmax')(x)

lenet = keras.Model(lenet_input,lenet_output,name='lenet')
lenet.summary()

lenet.compile(optimizer='SGD',
              loss='categorical_crossentropy',
              metrics=['accuracy'])

history=lenet.fit(train_images,
                  train_labels,
                  #batch_size=128,
                  epochs=20,
                  verbose=1,
                  validation_data=(test_images,test_labels))

plt.figure()
plt.plot(history.history['acc'],label='accuracy')
plt.plot(history.history['val_acc'],label='val_accuracy')
plt.xlabel('Epoch')
plt.ylabel('Accuracy')
plt.ylim([0.1,1])
plt.legend(loc='lower right')

标签：layers,plt,lenet5,labels,train,lenet,images,cv
来源： https://www.cnblogs.com/zhouyu0-0/p/11741262.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。