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)

这样运行后就可以打印出绝对路径了