ModuleNotFoundError: No module named 'torchvision.models.detection'

ModuleNotFoundError: No module named 'torchvision.models.detection'

今天在运行pytorch 的时候，出现了下面的错误：

Traceback (most recent call last): File "train.py", line 14, in from models.retinaface import RetinaFace File "/media/data/projects/faceDetect/Face-Detector-1MB-with-landmark/models/retinaface.py", line 3, in import torchvision.models.detection.backbone_utils as backbone_utilsModuleNotFoundError: No module named 'torchvision.models.detection'

解决方法

pip install torchvision==0.4

参考文献

[1].from torchvision.models.detection.faster_rcnn import FastRCNNPredictor builtins.ModuleNotFoundError: No module named 'torchvision.models'; 'torchvision' is not a package. ​​https://github.com/pytorch/vision/issues/1414​​

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。