importError: cannot import name 'model_urls' from 'torchvision.models.resnet'

 

Traceback (most recent call last):
File "/home/admin/PyTorch-Classification-Trainer/demo.py", line 17, in <module>
from classifier import inference
File "/home/admin/PyTorch-Classification-Trainer/classifier/inference.py", line 11, in <module>
from classifier.models.build_models import get_models
File "/home/admin/PyTorch-Classification-Trainer/classifier/models/build_models.py", line 13, in <module>
from torchvision.models.resnet import model_urls
ImportError: cannot import name 'model_urls' from 'torchvision.models.resnet' (/home/admin/miniconda3/envs/env311copy/lib/python3.11/site-packages/torchvision/models/resnet.py)

 

 

/home/admin/miniconda3/envs/env311copy/lib/python3.11/site-packages/torchvision/models/resnet.py  直接补上 model_urls 
 

model_urls = {
'resnet18': 'https://download.pytorch.org/models/resnet18-5c106cde.pth',
'resnet34': 'https://download.pytorch.org/models/resnet34-333f7ec4.pth',
'resnet50': 'https://download.pytorch.org/models/resnet50-19c8e357.pth',
'resnet101': 'https://download.pytorch.org/models/resnet101-5d3b4d8f.pth',
'resnet152': 'https://download.pytorch.org/models/resnet152-b121ed2d.pth',
}

参考：https://github.com/clovaai/CRAFT-pytorch/issues/191