执行提示上述错误。
root参数是根目录,不是具体的文件。
download参数,如果是True,则判断root目录下是否有文件,如果有,则解压该文件。如果没有,则去下载解压数据集文件到指定root目录。
解决方法:
方法1:
如果root目录没有文件或者放的下载的带解压的文件,则设置download=True。
方法2:
root目录下放已经解压好的文件,那么root设置为根目录,download设置为False即可。
代码示例:
import torchvision.datasets from torch.utils.tensorboard import SummaryWriter # 设置transforms方法 trans_tensor=torchvision.transforms.Compose( [torchvision.transforms.ToTensor()] ) train_data=torchvision.datasets.CIFAR10(root='./CIFAR',train=True,transform=trans_tensor,download=False) # test_data=torchvision.datasets.CIFAR10(root=r'./CIFAR',train=False,download=False) # print(train_data[0]) # img,target=train_data[0] # img.show() # print(img) # print(target) writer=SummaryWriter('dataset_logs') for i in range(15): img,target=train_data[i] writer.add_image('cifar',img,i) writer.close()