.vision和.utils的ModuleNotFoundError，download_url，VisionDataset

Question

我正在尝试在google colab中创建自定义数据集，但是导入给我错误。

from PIL import Image
from six.moves import zip
import os
from .vision import VisionDataset       ------------------------(1)
from .utils import download_url, check_integrity  --------------(2)

class datasetName(VisionDataset):
  ...

（1）错误：

  

ModuleNotFoundError：没有名为“ 主 .vision”的模块； “ 主要”是   不是包裹

（2）错误：

  

ModuleNotFoundError：没有名为“ 主 .utils”的模块； “ 主要”是   不是包裹

我尝试添加from torchvision import utils，但是它不能解决错误。

如果我更改为from torch.utils import download_url, check_integrity 然后错误变为：

  

ImportError：无法导入名称“ download_url”

Answer 1

请尝试以下导入行。

from torchvision.datasets.vision import VisionDataset
from torchvision.datasets.utils import download_url, check_integrity

希望有帮助！