yolov5導入模型出現ModuleNotyolov5導入模型出現ModuleNotFoundError: No module named ‘models‘FoundError: 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')

可以成功導入。

三. 總結
出現這種情況，可能跟當時模型保存的情況有關，例如訓練完的保存路徑包含’yolov5’字符串，如torch.save(‘yolov5/weights/yolov5s.pt’)
如果你的當前目錄是yolov5，則可以直接成功導入。

原文鏈接：https://blog.csdn.net/weixin_43508499/article/details/115536435