pytorch实现图像分类数据处理 1. pytorch库自带数据 数据预处理 数据生成 数据加载 2. 训练自己的数据 生成数据集 数据预处理
pytorch实现图像分类数据处理
- 1. pytorch库自带数据
- 数据预处理
- 数据生成
- 数据加载
- 2. 训练自己的数据
- 生成数据集
- 数据预处理
- 数据加载
1. pytorch库自带数据
为了更好的理解,这里以CIFAR10数据集作为训练和测试数据集。
我们将使用CIFAR10数据集,它包含十个类别:
[‘airplane’, ‘automobile’, ‘bird’, ‘cat’, ‘deer’, ‘dog’, ‘frog’, ‘horse’, ‘ship’, ‘truck’]。
CIFAR-10 中的图像尺寸为3x32x32,也就是RGB的3层颜色
通道,每层通道内的尺寸为32x32。
数据预处理