错误：导入错误：无法从“torchvision.models.vgg”导入名称“model_urls”

Question

我想使用 CRAFT 文本检测器模块。我安装了所有必需的软件包，但是当我尝试运行示例代码时，我得到了

    ImportError: cannot import name 'model_urls' from 'torchvision.models.vgg'

我寻找解决方案，但没有一个有效。

有什么建议吗？

Answer 1

显然

model_urls

已被删除，因为它在代码库中不一致 （参见 Github 上的评论）。

作者建议通过删除不存在模块的导入并获取模型 URL 信息来更新代码，如下所示：

import pkgutil
import torchvision
from importlib import import_module

def get_torchvision_models():
    model_urls = dict()
    for _, name, ispkg in pkgutil.walk_packages(torchvision.models.__path__):
        if ispkg:
            continue
        _zoo = import_module(f'torchvision.models.{name}')
        if hasattr(_zoo, 'model_urls'):
            _urls = getattr(_zoo, 'model_urls')
            model_urls.update(_urls)
    return model_urls

（来自 https://github.com/open-mmlab/mmcv/blob/fb486b96fd9932637a16f23a5dc60904c12bea7d/mmcv/runner/checkpoint.py#L108-L117）