CIFAR-10图像识别TensorFlow 2 实现

1、下载数据集

CIFAR-10是一个用于识别普适物 体的小型数据集，它包含了10个类 别的RGB彩色图片

import matplotlib.pyplot as plt
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers, models
import numpy as np

cifar10=tf.keras.datasets.cifar10
(Xtrain, Ytrain), (Xtest, Ytest) = cifar10.load_data()

2、数据预处理

Xtrain_normalize=Xtrain.astype("float32")/255.0
Xtest_normalize=Xtrain.astype("float32")/255.0

Ytrain_ohe=keras.utils.to_categorical(Ytrain)
Ytest_ohe=keras.utils.to_categorical(Ytest)

3、建立卷神经网络CNN模型

图像的特征提取：通过卷积层1，降采样层1，卷积层2以及降采样层2的处理，提取图像的特征

全连接神经网络：全连接层、输出层所组成的网络结构