每天五分钟深度学习框架pytorch:如何加载手写字体数据集mnist?
本文重点
那个这节课程之后,我们就将通过代码的方式来搭建CNN和RNN模型,然后训练,我们使用的数据集为pytorch中已经封装好的数据集,比如mnist,cafir10,本文我们学习一下如何在pytorch中使用它们,然后为之后的章节做准备,现在我们拿mnist来举例。
mnist和cafir10
MINST数据集是由Yann提供的手写数字数据库文件,这个数据集主要包含了60000张的训练图像和10000张的测试图像,它是的维度信息为[1,28,28],也就是说每张图片通道数是1,也就是灰度图,然后每张图片的大小是28*28
CAFIR10
cifar 10 这个数据集一共有 50000 张训练集,10000 张测试集,两个数据集里面的图片都是 png 彩色图片,图片大小是 3*32
原文地址:https://blog.csdn.net/huanfeng_AI/article/details/143610450
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!