深度学习项目一: 实现MNIST手写数字体的识别

小啊磊_Vv

已于 2023-06-28 09:51:03 修改

阅读量1k

点赞数

分类专栏：深度学习和视觉项目实战文章标签：人工智能计算机视觉 pytorch

于 2022-10-24 10:25:46 首次发布

本文链接：https://blog.csdn.net/m0_60890175/article/details/127485770

版权

深度学习和视觉项目实战专栏收录该内容

17 篇文章 7 订阅 ¥59.90 ¥99.00

订阅专栏

超级会员免费看

这篇博客介绍了如何使用PyTorch实现MNIST手写数字识别，涉及Vgg16_net和LeNet5两种网络结构。作者强调了数据预处理、网络构造以及训练过程，并提供了部分代码示例。

摘要由CSDN通过智能技术生成

实现MNIST手写数字体的识别

最近在找一些网络，想手动实现一下，于是就看到了MNIST数据集，于是找几个网络，自己试着实现一下。

前言

对于这个数据集输入是图像大小是 28*28，有时候所选的训练网络输入大小可能和MNIST数据集大小不一样，可能需要一些简单的操作对输入的大小进行简单的调整。
pytorch框架下进行训练。

一、数据集操作

代码如下（示例）：

导入相关的包
import torch
from torch import nn
import matplotlib.pyplot as plt
import torchvision
from torchvision.transforms import ToTensor
import torch.nn.functional as F
from torch.utils.data import DataLoader


# 数据集有关操作, MNIST是torchvision内置数据集