当前位置:   article > 正文

CIFAR10基于pytorch卷积神经网络(cnn)实战_cifar10_cnn_exercise.ipynb

cifar10_cnn_exercise.ipynb

 1、导入相关库

  1. import torch
  2. import torch.nn as nn #神经网络库
  3. import torchvision #下载数据集,图像转换
  4. from torchvision import transforms,datasets
  5. import torch.nn as nn
  6. import torch.nn.functional as F
  7. import torch.optim as optim

2、载入数据

  1. transform = transforms.Compose([ #定义transform
  2. transforms.ToTensor(),#将图片由numpy类型转换为tensor类型,torch中数据类型为张量,tensor类型
  3. transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))#归一化,0.5可自定义,范围在【-11】、
  4. ])
  5. trainset = datasets.CIFAR10(root="D:/cifar10",train=True,download=True,transform=transform)#下载训练集,root表示存放路径,train为训练集,trandform表示对数据进行转换
  6. testset = datasets.CIFAR10(root="D:/cifar10",train=False,download=True,transform=transform)#下载测试集,方法与训练集相

3、创建类

classes = {"plane","car","bird","cat","deer","dog","frog",&
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/46892
推荐阅读
相关标签
  

闽ICP备14008679号