PyTorch 卷积神经网络(CNN)

有关卷积神经网络的详细介绍,可参考:

PyTorch 卷积神经网络例子

本例将训练一个卷积神经网络,可以对图片(CIFAR-10)进行识别分类。

图片数据集使用CIFAR-10。CIFAR-10数据集包含10个类中的60000张32×32彩色图像,每个类包含6000张图像。训练图像50000张,测试图像10000张。

以下是数据集中的分类,以及每个类中的10张随机图像:

图

实现步骤

  1. 加载和标准化CIFAR10
  2. 定义卷积神经网络
  3. 定义损失函数
  4. 使用训练数据对网络进行训练
  5. 用测试数据测试网络

章节



浙ICP备17015664号-1 浙公网安备 33011002012336号 联系我们 网站地图  
@2019 qikegu.com 版权所有,禁止转载