执行提示上述错误。

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()

 

 posted on 2024-03-08 18:35  会飞的金鱼  阅读(262)  评论(0)    收藏  举报