卷积神经网络，TensorFlow实现

VIP文章「已注销」

已于 2022-09-20 10:07:01 修改

阅读量328

点赞数 1

分类专栏：深度学习文章标签： tensorflow 深度学习 python

于 2022-09-20 09:59:29 首次发布

本文链接：https://blog.csdn.net/m0_73817310/article/details/126947102

版权

import tensorflow as tf
import matplotlib.pyplot as plt
from tensorflow import keras
from tensorflow.keras import datasets
from tensorflow.keras import layers
print("TensorFlow version:", tf.__version__)

TensorFlow version: 2.6.4

下载并准备 CIFAR10 数据集

CIFAR10 数据集包含 10 类，共 60000 张彩色图片，每类图片有 6000 张。此数据集中 50000 个样例被作为训练集，剩余 10000 个样例作为测试集。类之间相互独立，不存在重叠的部分。

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

# Normalize pixel values to be between 0 and 1
train_images, test_images = train_images / 255.0, test_images / 255.0

Downloading data from https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
170500096/170498071 [==============================] - 6s 0us/step
170508288/170498071 [==============================] - 6s 0us/step

构造卷积神经网络模型

inputs = keras.Input(shape=(32, 32, 3))
x = layers.Conv2D(filters=32, kernel_size=3, activation="relu")(inputs)
x = layers.MaxPooling2D(pool_size=2)(x)
x = layers.