1.问题
当我们下载torchvision.datasets里面的数据集时,有时候会遇到找不到数据集保存路径的问题。
2.解决
引入os库
import os
调用如下方法
os.path.abspath(数据集对象.root)
以下面代码为例
import osimport torchvision.datasets as datasets# 指定数据集的根路径root = '/data'# 获取CIFAR-10数据集的保存路径train_dataset = datasets.CIFAR10(root=root, train=True, download=True)test_dataset = datasets.CIFAR10(root=root, train=False, download=True)train_dataset_path = os.path.abspath(train_dataset.root)test_dataset_path = os.path.abspath(test_dataset.root)print("训练集保存绝对路径:", train_dataset_path)print("测试集保存绝对路径:", test_dataset_path)
这样运行后就可以打印出绝对路径了