用pytorch加载自己的数据集(使用Dataloader替代Torchvision.datasets.MNIST)(附项目源码)
参考:https://blog.csdn.net/winycg/article/details/90318371省流:当使用dataloader函数切割数据集与标签时,可以以列表[元组(tensor数据,int标签)]的格式输入。例:将文件夹中的1000张图片输入代码,每张图片尺寸1*28*28,标签以int格式输入。分割批次大小:128#数据:torch.Size([1,28,28])|图像#标