CIFAR-10 是机器学习领域经典的图像数据集,包含 10 个类别(如飞机、汽车、鸟等),共 60,000 张 32x32 彩色图片。使用 TensorFlow 处理该数据集是入门深度学习的绝佳实践,以下是关键步骤:
📦 数据加载与预处理
加载数据
使用tf.keras.datasets.cifar10.load_data()
获取数据,数据会自动分为训练集和测试集CIFAR-10 样本数据标准化
将像素值归一化到 [0,1] 范围:x_train = x_train.astype('float32') / 255 x_test = x_test.astype('float32') / 255
数据增强(可选)
使用ImageDataGenerator
添加旋转、平移等变换:datagen = ImageDataGenerator(rotation_range=10, horizontal_flip=True)
🛠️ 模型构建示例
使用 TensorFlow 构建简单 CNN 模型:
model = Sequential([
Conv2D(32, (3,3), activation='relu', input_shape=(32,32,3)),
MaxPooling2D(2,2),
Flatten(),
Dense(10, activation='softmax')
])
📈 训练与评估
- 训练命令:
model.fit(x_train, y_train, epochs=10, validation_split=0.2)
- 评估指标:准确率、损失值训练曲线
🌐 扩展学习
想深入了解 TensorFlow 的高级用法?
点击这里查看 TensorFlow 官方教程
📌 提示:CIFAR-10 数据集可通过 此路径 下载完整文件