google-colaboratory - .vision 和 .utils、download_url、VisionDataset 的 ModuleNotFoundError

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：没有名为“ main .vision”的模块；' main ' 不是一个包

(2) 错误：

ModuleNotFoundError：没有名为“ main .utils”的模块；' main ' 不是一个包

我试图添加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

希望能帮助到你！