使用CRAFT文本检测器遇报错：无法从torchvision.models.vgg导入model_urls

• 修改CRAFT代码中的导入逻辑
  找到CRAFT代码里导入model_urls的文件（通常是加载VGG模型的模块，比如craft_utils.py或自定义的vgg.py），将原有导入语句：

  from torchvision.models.vgg import model_urls
  

  替换为以下两种方案之一：

  # 方案1：从torchvision内部工具模块导入对应变量
  from torchvision.models._utils import _VGG_MODEL_URLS as model_urls
  # 方案2：手动定义所需的VGG模型URL
  model_urls = {
      'vgg16': 'https://download.pytorch.org/models/vgg16-397923af.pth',
  }
  

• 降低torchvision版本适配
  若不想修改代码，可安装兼容的旧版本torchvision（对应torch版本需匹配），例如：

  pip install torchvision==0.12.0 torch==1.12.0 --force-reinstall
  

  注：需根据本地CUDA版本调整torch的具体版本号，保证两者版本兼容。

• 替换预训练模型加载方式
  找到CRAFT中使用model_urls加载权重的代码，改为直接加载本地预训练权重文件：

  # 原代码示例
  state_dict = load_state_dict_from_url(model_urls['vgg16'])
  # 替换为本地加载
  state_dict = torch.load('/your/local/path/vgg16.pth')
  

内容的提问来源于stack exchange，提问作者cobdmg