官网文档链接:
https://pytorch.org/vision/stable/datasets.html
截图
数据集MNIST
train-images-idx3-ubyte:训练集图像
train-labels-idx1-ubyte:训练集标签
t10k-images-idx3-ubyte:测试集图像
t10k-labels-idx1-ubyte:测试集标签
- root:训练集图像和测试集图像数据集的根目录
- train:如果train=True那么数据集来自训练集,否则来自训练集
- download:如果download=True,那么从互联网上下载数据集并放到根目录,如果数据集已经下载了,那么就不再下载
- transform :一種函數/轉換,用於接收 PIL 圖像並返回轉換后的版本
原创文章,作者:端木书台,如若转载,请注明出处:https://blog.ytso.com/278432.html