CIFAR10识别

1. 数据集简介

CIFAR10数据集共有6W张彩色图像,图像大小是32*32*3的,共计10个类,每类6K张图片。

其中训练集5W张,构成了5个训练批,每一批1W张,但一个训练批中的各类图像并不一定数量相同,总的来看训练集,每一类都有5K张;测试集1W张单独构成一批,其来自10个分类,每类随机取1K张。

CIFAR10识别_第1张图片

2. 数据加载

2.1 数据集下载

只下载一次, 批量迭代读取

def load_data(batch_size):
    # 1. 构建数据转换器,进行数据增强
    transform = get_transform()

    # 2. 下载数据集
    train_set = datasets.CIFAR10(root='../data/', train=True, download=True, transform=transform)
    test_set = datasets.CIFAR10(root='../data/', train=False, download=True, transform&#

你可能感兴趣的:(pytorch,深度学习,cnn,计算机视觉)