PyTorch ------AlexNet卷积神经网络实现mnist手写体识别

  • 接上一篇LeNet现实mnist手写识别

  • 使用经典模型AlexNet实现相同的功能

  • 先简单介绍一下,今天的主角,重燃神经网络辉煌的的模型AlexNet

AlexNet是2012年ImageNet竞赛冠军获得者Hinton和他的学生设计的,在那年之后,更多的更深的神经网络如雨后春笋般出现,比如优秀的vgg,GoogLeNet等。 对于传统的机器学习分类算法而言,已经是相当出色。

  • 简单介绍一下 AlexNet模型的结构:
    • 结构如下如:
      -在这里插入图片描述借用百度百科的图片展示.
    • 使用的新技术如下:
      • 8层变换,其中有5层卷积和2层全连接隐藏层,以及1个全连接输出层。
      • 将sigmoid激活函数改成了更加简单的ReLU激活函数。
      • 用Dropout来控制全连接层的模型复杂度。
      • 使用CUDA加速深度卷积网络的训练
      • 引入数据增强,如翻转、裁剪和颜色变化,从而进一步扩大数据集来缓解过拟合
    • 代码兼容GPU
    • 这个模型比较复杂,在CPU上很是费劲,尽可能在GPU上运行.
    • 代码中包含可训练参数的计算、features Map的大小计算
    • 代码中备注又不清楚的请留言,我们要做一个严谨的Coder
    • 不能运行的都是耍流氓
    • 下面上代码:

      #引入使用库
      import time
      import torch
      from torch import nn,optim
      import torchvision
  • 6
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
下面是一个简单的 PyTorch 手写体识别示例代码: ```python import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import DataLoader, TensorDataset from torchvision.datasets import MNIST from torchvision.transforms import ToTensor # 加载数据集 train_dataset = MNIST(root='data', train=True, transform=ToTensor(), download=True) test_dataset = MNIST(root='data', train=False, transform=ToTensor(), download=True) # 构建数据加载器 train_loader = DataLoader(train_dataset, batch_size=64, shuffle=True) test_loader = DataLoader(test_dataset, batch_size=64, shuffle=False) # 定义模型 class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv = nn.Sequential( nn.Conv2d(1, 32, kernel_size=3, padding=1), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2), nn.Conv2d(32, 64, kernel_size=3, padding=1), nn.ReLU(), nn.MaxPool2d(kernel_size=2, stride=2), ) self.fc = nn.Sequential( nn.Linear(64 * 7 * 7, 128), nn.ReLU(), nn.Linear(128, 10), ) def forward(self, x): x = self.conv(x) x = x.view(x.size(0), -1) x = self.fc(x) return x # 定义模型、损失函数和优化器 model = Net() criterion = nn.CrossEntropyLoss() optimizer = optim.Adam(model.parameters(), lr=0.001) # 训练模型 for epoch in range(10): running_loss = 0.0 for i, (inputs, labels) in enumerate(train_loader, 0): optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() # 打印训练信息 print('Epoch %d loss: %.3f' % (epoch+1, running_loss/len(train_loader))) # 测试模型 correct = 0 total = 0 with torch.no_grad(): for inputs, labels in test_loader: outputs = model(inputs) _, predicted = torch.max(outputs.data, 1) total += labels.size(0) correct += (predicted == labels).sum().item() # 打印测试结果 print('Accuracy on %d test images: %.2f %%' % (total, 100 * correct / total)) ``` 在这个示例代码中,我们首先使用 PyTorch 内置的 MNIST 数据集下载工具加载了训练集和测试集,并构建了数据加载器。然后,我们定义了一个包含两个卷积层和两个全连接层的卷积神经网络,并使用交叉熵损失函数和 Adam 优化器进行模型训练。最后,我们使用测试集对模型进行测试,并计算出了模型在测试集上的准确率。 这个示例代码只是一个简单的手写体识别PyTorch 实现,你可以根据自己的需求进行修改和扩展。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值