当前位置:   article > 正文

实战2:基于卷积cnn得图像分类详细教程_cnn图像分类实战项目

cnn图像分类实战项目
  1. #配置库
  2. import torch
  3. from torch import nn,optim
  4. import torch.nn.functional as F
  5. from torch.autograd import Variable
  6. from torch.utils.data import DataLoader
  7. from torchvision import transforms
  8. from torchvision import datasets
  9. import matplotlib.pyplot as plt
  10. import numpy as np
  11. import torch.nn as nn
  12. import torch.nn.functional as F
  13. import torch.optim as optim # 优化器
  14. import torchvision
  15. me v 15563370935
  16. #配置参数
  17. torch.manual_seed(1)#设置随机数种子,确保结果可重复
  18. # batch_size=128 #批处理大小
  19. learning_rate=0.001 #学习率
  20. num_epoches=10 #训练次数
  21. #加载CIFAR-10数据
  22. #(ToTensor():把一个PIL.Image转换成Tensor,Normalize():标准化,即减均值,除以标准差)
  23. transform=transforms.Compose([transforms.ToTensor(),transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
  24. #训练集
  25. trainset=datasets.CIFAR10(root='./data', train=True,download=True, transform=transform)
  26. #测试集
  27. testset=datasets.CIFAR10(root='./data', train=False,download=True, tra
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/721154
推荐阅读
相关标签
  

闽ICP备14008679号