main(config) File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 24, in main from models import build_model, build_loss File "/home/ps/wangh.._moduleno">
赞
踩
File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 78, in <module>
main(config)
File "/home/ps/wanghao/DBNet.pytorch-master/tools/train.py", line 24, in main
from models import build_model, build_loss
File "/home/ps/wanghao/DBNet.pytorch-master/models/__init__.py", line 5, in <module>
from .model import Model
File "/home/ps/wanghao/DBNet.pytorch-master/models/model.py", line 8, in <module>
from models.backbone import build_backbone
File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/__init__.py", line 7, in <module>
from .shufflenetv2 import *
File "/home/ps/wanghao/DBNet.pytorch-master/models/backbone/shufflenetv2.py", line 7, in <module>
from torchvision.models.utils import load_state_dict_from_url
ModuleNotFoundError: No module named 'torchvision.models.utils'
将
from torchvision.models.utils import load_state_dict_from_url
改为:
from torch.hub import load_state_dict_from_url
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。