Pythorch的数据加载和处理教程是非常具体的一个例子,有人可以帮助我什么样的功能应该像一个更通用的简单图像加载?
教程:http://pytorch.org/tutorials/beginner/data_loading_tutorial.html
我的数据:
我将MINST数据集作为jpg保存在下面的文件夹结构中。(我知道我可以只使用dataset类,但这纯粹是为了了解如何将简单的图像加载到pytorch中,而不使用csv或复杂的特性)。
文件夹名是标签,图像是28x28png的灰度,不需要转换。
data
train
0
3.png
5.png
13.png
23.png
...
1
3.png
10.png
11.png
...
2
4.png
13.png
...
3
8.png
...
4
...
5
...
6
...
7
...
8
...
9
...
如果您正在使用mnist,那么pytorch中已经通过torchvision设置了一个预设值。
你可以的
如果您想泛化到图像目录(与上面的导入相同),可以
然后,您可以对其进行迭代,如下所示:
有很多方法可以推广pytorch用于图像数据集加载,我知道的方法是子类化torch.utils.data.dataset
下面是我为Pythorch0.4.1所做的(应该仍然在1.3中工作)
相关问题 更多 >
编程相关推荐