'datasets.ImageFolder' 在 python 中不起作用

Question

当我运行这段代码时，我得到了这个错误： NameError: 名称“数据集”未定义

我该如何解决这个问题？ 非常感谢

import numpy as np
import torch
from torchvision import transforms
from torchvision.datasets.folder import default_loader
np.random.seed(random_seed)
torch.manual_seed(random_seed)


train_dataset = datasets.ImageFolder("/content/gdrive/My Drive/val2017", 
transform)

Answer 1

如果您使用的是 anaconda 发行版，请首先使用以下命令安装 torchvision：

$ conda install -c conda-forge torchvision

然后

from torchvision import datasets