Pytorch利用lstm做mnist手写数字识别分类

 以下是模型的运行结果:

MNIST共包含70000张手写数字图片,其中有60000张用作训练集,10000张用作测试集。原始数据集可在MNIST官网下载

数据集包含了0-9共10类手写数字图片,每张图片都做了尺寸归一化,都是28x28大小的灰度图

pytorch 利用lstm做mnist手写数字识别分类

将MNIST数据集保存成.jpg图片格式

mnist2jpg.py

# coding=utf-8
'''将二进制格式的MNIST数据集转成.jpg图片格式并保存,图片标签包含在图片名中'''
import numpy as np
import c

猜你喜欢

转载自blog.csdn.net/tony2278/article/details/105200963
今日推荐