PyTorch实现一个简单的图像分类(代码详细)

GitHub源码

B站视频

关注我的小伙伴都知道,在很早以前就已经开始写有关深度学习的文章,但是为什么今天还要写一个大家都“听腻”的话题呢!也就是图像分类。我相信,对于大部分在学习深度学习的小伙伴来说,这个应该是一个很好理解的问题,因为图像分类作为很多算法模型的基础,大家都觉得很简单,虽然简单,但是想要提出一个好的算法很难,如今深度学习火爆已有10年了,为什么现在很多领域算法模型都是基于图像分类模型来的呢或者对其进行改进,思想上并没有太多本质的区别,这也说明了图像分类算法的重要性以及对其一个更加深度理解的重要性。

我也首次在B站展示了一个简单的图像分类算法模型以及代码的复现过程,主要是想从基础的开始理解,带着大家实现代码,虽然没有去详细讲解分类算法,但是对于大家在理解和实现的过程还是挺重要的。

1.本文以手写体数字识别数据集;

2.根据自己搭建的网络模型复现整个代码;

3.训练模型,验证模型。

使用pytorch实现图像分类代码如下:import torch import torchvision import torchvision.transforms as transforms# 定义数据加载器 # 载入CIFAR10数据集 transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, shuffle=True, num_workers=2)# 定义模型 import torch.nn as nn import torch.nn.functional as Fclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2, 2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16 * 5 * 5, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 10) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) x = x.view(-1, 16 * 5 * 5) x = F.relu(self.fc1(x)) x = F.relu(self.fc2(x)) x = self.fc3(x) return x# 定义损失函数和优化器 import torch.optim as optimnet = Net() criterion = nn.CrossEntropyLoss() optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)# 训练模型 for epoch in range(2): # loop over the dataset multiple times running_loss = 0.0 for i, data in enumerate(trainloader, 0): # get the inputs; data is a list of [inputs, labels] inputs, labels = data # zero the parameter gradients optimizer.zero_grad() # forward + backward + optimize outputs = net(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() # print statistics running_loss += loss.item() if i % 2000 == 1999: # print every 2000 mini-batches print('[%d, %5d] loss: %.3f' % (epoch + 1, i + 1, running_loss / 2000)) running_loss = 0.0print('Finished Training')
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值