解决yolov8推理报错:ModuleNotFoundError: No module named ‘ultralytics.nn.modules.conv‘

1.项目场景:

在GitHub上下载了一个yolov8的代码,训练好检测模型,运行作者给的推理代码报错:

python 复制代码
Traceback (most recent call last):
  File "E:\Pycharm\pytorch-project\YOLOv8\inference.py", line 9, in <module>
    model = YOLO("./runs/detect/train/weights/last.pt")
  File "E:\Pycharm\pytorch-project\YOLOv8\ultralytics\yolo\engine\model.py", line 55, in __init__
    {'.pt': self._load, '.yaml': self._new}[Path(model).suffix](model)
  File "E:\Pycharm\pytorch-project\YOLOv8\ultralytics\yolo\engine\model.py", line 83, in _load
    self.model, self.ckpt = attempt_load_one_weight(weights)
  File "E:\Pycharm\pytorch-project\YOLOv8\ultralytics\nn\tasks.py", line 341, in attempt_load_one_weight
    ckpt = torch.load(attempt_download(weight), map_location='cpu')  # load
  File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\serialization.py", line 809, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\serialization.py", line 1172, in _load
    result = unpickler.load()
  File "E:\Anaconda3\envs\pytorch\lib\site-packages\torch\serialization.py", line 1165, in find_class
    return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'ultralytics.nn.modules.conv'; 'ultralytics.nn.modules' is not a package

2.原因分析:

作者仓库地址为:https://github.com/DataXujing/YOLOv8

首先看错误原因:'ultralytics.nn.modules'is not a package,说明ultralytics.nn模块有问题。

查看作者ultralytics包的内容,并与最新版本对比:

8.0.0:

8.0.229:

可以看出,两者的modules不同


3.解决方案:

解决办法很简单,把下载下来的nn模块替换掉即可,问题解决。

相关推荐
羊羊小栈3 小时前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业
大霸王龙1 天前
基于vLLM与YOLO的智能图像分类系统
yolo·分类·数据挖掘
m_136872 天前
Mac M 系列芯片 YOLOv8 部署教程(CPU/Metal 后端一键安装)
yolo·macos
格林威2 天前
机器视觉在半导体制造中有哪些检测应用
人工智能·数码相机·yolo·计算机视觉·视觉检测·制造·相机
、、、、南山小雨、、、、2 天前
YOLO在ubuntu22安装
yolo
羊羊小栈3 天前
基于「YOLO目标检测 + 多模态AI分析」的铁路轨道缺陷检测安全系统(vue+flask+数据集+模型训练)
人工智能·yolo·目标检测·语言模型·毕业设计·创业创新·大作业
Python图像识别3 天前
63_基于深度学习的草莓病害检测识别系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
范男3 天前
YOLO11目标检测运行推理简约GUI界面
图像处理·人工智能·yolo·计算机视觉·视觉检测
model20053 天前
ubuntu24.04+5070ti训练yolo模型(2)
人工智能·yolo
强盛小灵通专卖员4 天前
RK3576边缘计算设备部署YOLOv11
人工智能·深度学习·yolo·边缘计算·sci·rk3576·小论文