YOLOv8模型验证以及一些报错

学yolo的小白

已于 2024-04-09 20:51:34 修改

阅读量2.3k

点赞数 11

分类专栏： upgradeYOLOv8 Upgrade YOLOv8进阶文章标签： YOLO python

于 2024-03-12 15:09:35 首次发布

本文链接：https://blog.youkuaiyun.com/zmyzcm/article/details/136652541

版权

Upgrade YOLOv8进阶同时被 2 个专栏收录

25 篇文章 ¥39.90 ¥99.00

订阅专栏

upgradeYOLOv8

13 篇文章

订阅专栏

本文主要介绍了如何验证YOLOv8模型的添加成功，以及在训练过程中遇到的KeyError问题的解决方法，该错误由虚拟环境中软件包版本不一致引起。此外，还讨论了由于YOLOv8版本差异导致的TypeError报错及其解决方案，包括使用新版本博主的改进或切换到旧版本源码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、验证模型是否添加成功

在根目录下新建一个py文件添加代码：

from ultralytics import  YOLO
from ultralytics.utils import ASSETS

CFG = 'xx/ultralytics/ultralytics/cfg\models/v8/yolov8+XX.yaml'  ##改进的yaml文件位置##
SOURCE = ASSETS / 'bus.jpg'

def test_model_forward():
    """Test the forward pass of the YOLO model."""
    model = YOLO(CFG)
    model(source=None, imgsz=32, augment=True)

运行验证。

二、一些报错

1、当上边代码成功跑通，模型已经添加了改进后的结构，但是训练的时候报错“KeyError:XXX”

是因为跑模型所用的虚拟环境里也有一个ultralytics的软件包，在项目Yolov8里改动的代码，虚拟环境里并没有改，所以训练时候会出现这种找不到key的报错。

解决办法，找到报错的虚拟环境路径一般是anaconda3\envs\yolov8（你的环境名字）\Lib\site-packages\ultralytics-x.x.x-pyx.x.egg/ultralytics,把这个路径中的ultralytics文件夹替换成在pycharm中修改后的ultralytics文件夹。

二、yolov8版本不同出现报错

作者所有博客中的改进都是在23年11月左右的yolov8版本中进行的，与现在github上的源码有些许不同。

出现TypeError: _predict_once() takes from 2 to 4 positional arguments but 5 were given或者TypeError: list indices must be integers or slices, not list等可能是因为版本不同出现的报错。

解决方法：1找别的博主基于新版本的yolov8改进博客改进，

或者2下载老版本的yolov8源码试试。

地址：【免费】2023.11.15版本GitHubYOLOv8源码资源-优快云文库