卷积层:可以认为是利用图像的多个滤波器(经常被称为卷积操作)进行滤波,得到图像的特征。 通常,我们在 PyTorch 中使用nn.Conv2d定义卷积层,并指定以下参数: nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0) in_channels是指输入深度。对于灰阶图像来说,深度 = 1 out_channels是指输出深度,...
所以选用CIFAR-10数据集实现图片的分类。 1. CIFAR-10数据集简介 CIFAR-10数据集由60000个32x32彩色图像组成,包括10个类别,每个类别有6000个图像。有50000个训练图像和10000个测试图像。 数据集分为五个训练批次和一个测试批次,每个批次有10000个图像。测试批次包含来自每个类的正好1000个随机选择的图像。训练批次以...