当前位置：首页 > ai >正文

TensorFlow 2.x入门实战：从零基础到图像分类项目

ai 2025/7/1 14:00:48

TensorFlow 2.x入门实战：从零基础到图像分类项目

前言

TensorFlow是Google开发的开源机器学习框架，已成为深度学习领域的重要工具。TensorFlow 2.x版本相比1.x有了重大改进，更加易用且功能强大。本文将带你从零开始学习TensorFlow 2.x，最终完成一个图像分类项目。

第一部分：TensorFlow 2.x基础

1. TensorFlow 2.x安装

首先确保你已安装Python 3.6-3.8，然后使用pip安装TensorFlow：

pip install tensorflow
# 或者安装GPU版本（需要CUDA和cuDNN）
pip install tensorflow-gpu

2. TensorFlow核心概念

张量(Tensor): TensorFlow中的基本数据类型，可以看作是多维数组。

import tensorflow as tf# 创建张量
scalar = tf.constant(3.0)          # 标量（0维张量）
vector = tf.constant([1, 2, 3])    # 向量（1维张量）
matrix = tf.constant([[1, 2], [3, 4]])  # 矩阵（2维张量）
tensor = tf.constant([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])  # 3维张量

变量(Variable): 用于存储模型参数的可变张量。

weights = tf.Variable(tf.random.normal([3, 2]))
bias = tf.Variable(tf.zeros([2]))

自动微分(GradientTape): TensorFlow 2.x使用GradientTape记录计算过程以便自动求导。

x = tf.Variable(3.0)
with tf.GradientTape() as tape:y = x**2
dy_dx = tape.gradient(y, x)  # dy/dx = 6.0

第二部分：构建第一个神经网络

1. 使用Keras API构建模型

TensorFlow 2.x集成了Keras API，使模型构建更加简单。

from tensorflow.keras import layers, models# 构建序列模型
model = models.Sequential([layers.Dense(64, activation='relu', input_shape=(784,)),layers.Dense(64, activation='relu'),layers.Dense(10, activation='softmax')
])# 编译模型
model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'])

2. 训练模型

# 加载数据（这里以MNIST为例）
mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0  # 归一化# 训练模型
history = model.fit(x_train.reshape(-1, 784), y_train, epochs=5, validation_split=0.2)# 评估模型
test_loss, test_acc = model.evaluate(x_test.reshape(-1, 784), y_test)
print(f"Test accuracy: {test_acc}")

第三部分：图像分类项目实战

我们将使用CIFAR-10数据集构建一个卷积神经网络(CNN)进行图像分类。

1. 数据准备

# 加载CIFAR-10数据集
(train_images, train_labels), (test_images, test_labels) = tf.keras.datasets.cifar10.load_data()# 归一化像素值到0-1之间
train_images, test_images = train_images / 255.0, test_images / 255.0# 类别名称
class_names = ['airplane', 'automobile', 'bird', 'cat', 'deer','dog', 'frog', 'horse', 'ship', 'truck']

2. 构建CNN模型

model = models.Sequential([layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)),layers.MaxPooling2D((2, 2)),layers.Conv2D(64, (3, 3), activation='relu'),layers.MaxPooling2D((2, 2)),layers.Conv2D(64, (3, 3), activation='relu'),layers.Flatten(),layers.Dense(64, activation='relu'),layers.Dense(10)
])model.compile(optimizer='adam',loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=['accuracy'])

3. 训练与评估

history = model.fit(train_images, train_labels, epochs=10, validation_data=(test_images, test_labels))# 评估模型
test_loss, test_acc = model.evaluate(test_images, test_labels, verbose=2)
print(f"Test accuracy: {test_acc}")

4. 模型优化与改进

为了提高模型性能，我们可以添加数据增强和Dropout层：

# 数据增强
data_augmentation = tf.keras.Sequential([layers.experimental.preprocessing.RandomFlip("horizontal"),layers.experimental.preprocessing.RandomRotation(0.1),layers.experimental.preprocessing.RandomZoom(0.1),
])# 改进后的模型
model = models.Sequential([data_augmentation,layers.Conv2D(32, (3, 3), activation='relu', input_shape=(32, 32, 3)),layers.MaxPooling2D((2, 2)),layers.Conv2D(64, (3, 3), activation='relu'),layers.MaxPooling2D((2, 2)),layers.Conv2D(128, (3, 3), activation='relu'),layers.Flatten(),layers.Dropout(0.5),layers.Dense(128, activation='relu'),layers.Dense(10)
])model.compile(optimizer='adam',loss=tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True),metrics=['accuracy'])# 训练更多epoch
history = model.fit(train_images, train_labels, epochs=30, validation_data=(test_images, test_labels))

第四部分：模型保存与部署

1. 保存模型

# 保存整个模型
model.save('cifar10_model.h5')# 或者只保存权重
model.save_weights('cifar10_weights.ckpt')

2. 加载模型进行预测

# 加载模型
loaded_model = tf.keras.models.load_model('cifar10_model.h5')# 对新数据进行预测
predictions = loaded_model.predict(test_images[:5])
predicted_classes = tf.argmax(predictions, axis=1)
print([class_names[i] for i in predicted_classes])