CNN 手写数字识别 (pytorch代码实现)

最新推荐文章于 2024-12-27 12:07:51 发布

_missTu_

最新推荐文章于 2024-12-27 12:07:51 发布

阅读量1.9k

点赞数 2

分类专栏： pyTorch

本文链接：https://blog.csdn.net/u013075024/article/details/106302942

版权

1、CNN 手写数字识别步骤

（1）加载minist数据
（2）搭建网络
（3）模型训练
（4）结果可视化

2、pytorch代码实现

# library
# standard library
import os

# third-party library
import torch
import torch.nn as nn
import torch.utils.data as Data
import torchvision
import matplotlib.pyplot as plt

# torch.manual_seed(1)    # reproducible

# Hyper Parameters
EPOCH = 1               # train the training data n times, to save time, we just train 1 epoch
BATCH_SIZE = 50
LR = 0.001              # learning rate
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_data = torchvision.datasets.MNIST(
    root='./mnist/',
    train=True,                                     # this is training data
    tr