yolov5导入模型出现ModuleNotFoundError: No module named ‘models‘

一. 问题

现有文件目录如下所示：

现在在这层目录下导入模型：

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')

可以成功导入。

三. 总结

1. 出现这种情况，可能跟当时模型保存的情况有关，例如训练完的保存路径包含’yolov5’字符串，如torch.save(‘yolov5/weights/yolov5s.pt’)
2. 如果你的当前目录是yolov5，则可以直接成功导入。