之前讲的例子,程序都是调用的datasets方法,下载的torchvision本身就提供的数据,那么如果想导入自己的数据应该怎么办呢?
本篇就讲解一下如何创建自己的数据集。
还有第二篇……Pytorch打怪路(三)Pytorch创建自己的数据集2
1.用于分类的数据集
以mnist数据集为例
这里的mnist数据集并不是torchvision里面的,而是我自己的以图片格式保存的数据集,因为我在测试STN时,希望自己再把这些手写体做一些形变,
所以就先把MNIST数据集转化成了jpg图片格式,然后做了一些形变,当然这不是重点。首先我们看一下我的数据集的情况:
如图所示,我的图片数据集确实是jpg图片
再看我的存储图片名和label信息的文本: