yolov5导入模型出现ModuleNotFoundError: No module named ‘models‘
目录一. 问题二. 解决方案三. 总结一. 问题现有文件目录如下所示:现在在这层目录下导入模型:import torchmodel = torch.load('yolov5/yolov5s.pt')出现问题:二. 解决方案在导入模型前,把yolov5的路径insert到sys.pathimport torchimport syssys.path.insert(0, './yolov5')model
·
一. 问题
现有文件目录如下所示:
现在在这层目录下导入模型:
import torch
model = torch.load('yolov5/yolov5s.pt')
出现问题:
二. 解决方案
在导入模型前,把yolov5的路径insert到sys.path
import torch
import sys
sys.path.insert(0, './yolov5')
model = torch.load('yolov5/yolov5s.pt')
可以成功导入。
三. 总结
- 出现这种情况,可能跟当时模型保存的情况有关,例如训练完的保存路径包含’yolov5’字符串,如torch.save(‘yolov5/weights/yolov5s.pt’)
- 如果你的当前目录是yolov5,则可以直接成功导入。
更多推荐
已为社区贡献3条内容
所有评论(0)