pytorch 卷积神经网络CNN实例——手写数字识别

最新推荐文章于 2024-10-04 16:38:46 发布

Kalankalan

最新推荐文章于 2024-10-04 16:38:46 发布

阅读量4.5k

点赞数 2

分类专栏：深度学习文章标签： CNN实例卷积神经网络

本文链接：https://blog.csdn.net/u014556057/article/details/85173966

版权

通过PyTorch构建卷积神经网络（CNN）模型，实例演示如何使用MNIST数据集进行手写数字识别。参考代码包含了自动下载数据集的流程。

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

数据集官网：http://yann.lecun.com/exdb/mnist/
参考：https://github.com/MorvanZhou/PyTorch-Tutorial/blob/master/tutorial-contents/401_CNN.py

代码包含自动下载数据集

import torch
from torch import nn,optim
import torch.nn.functional as F
from torch.autograd import Variable
from torch.utils.data import DataLoader
from torchvision import transforms
from torchvision import datasets
import os


batch_size = 200    #分批训练数据、每批数据量
learning_rate = 1e-2    #学习率
num_epoches = 20       #训练次数
DOWNLOAD_MNIST = False    #是否网上下载数据

# Mnist digits dataset
if not(os.path.exists('./mnist/')) or not os.listdir('./mnist/'):
    # not mnist dir or mnist is empyt dir
    DOWNLOAD_MNIST = True

train_dataset = datasets.MNIST(
    root = './mnist',
    train= True,