简单的卷积神经网络实现

卷积神经网络

参考教程：CNN（莫烦Python19）

卷积神经网络流程图

实现代码

import torch
import torch.nn as nn  #神经网络的层
import torch.utils.data as Data #批处理数据
import torchvision #包括数据库
import matplotlib.pyplot as plt
from torch.autograd import Variable

#hyper parameters
EPOCH = 1
BATCH_SIZE = 50
LR = 0.001
DOWNLOAD_MNIST = False

#训练集
train_data = torchvision.datasets.MNIST(
    root = './mnist', #数据保存的位置
    train = True, #用的训练集的数据，False表示测试集
    transform=torchvision.transforms.ToTensor(), #把下载的数据改为tensor的格式,数据值为（0,1）
    download=DOWNLOAD_MNIST
)


#plot one example
# print(train_data.train_data.size()) #(60000, 28, 28)
# print(train_data.train_labels.size()) #(60000)
# plt.imshow(train_data.train_data[0].numpy(), cmap='gray')
# plt.title('%i'%train_data.train_labels[0])
# plt.show()

#数据批处理
train_loader = Data.DataLoader(
    dataset = train_data,