YOLOv8模型验证以及一些报错

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

在根目录下新建一个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的报错。

解决办法,找到报错的虚拟环境路径一般是xx\xx\lib\site-packages\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源码资源-CSDN文库

相关推荐

  1. YOLOv8模型验证以及一些

    2024-03-14 15:16:02       44 阅读
  2. 一些k8s的小的记录

    2024-03-14 15:16:02       61 阅读
  3. yolov8模型转换

    2024-03-14 15:16:02       47 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-03-14 15:16:02       94 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-03-14 15:16:02       100 阅读
  3. 在Django里面运行非项目文件

    2024-03-14 15:16:02       82 阅读
  4. Python语言-面向对象

    2024-03-14 15:16:02       91 阅读

热门阅读

  1. 代码随想录day39:动态规划part7,完全背包继续

    2024-03-14 15:16:02       37 阅读
  2. django根据时间(年月日)动态修改表名--方法二

    2024-03-14 15:16:02       40 阅读
  3. ES6 Nunber类型、Math对象扩展

    2024-03-14 15:16:02       38 阅读
  4. IOS面试题object-c 81-90

    2024-03-14 15:16:02       42 阅读
  5. mysql数据库备份恢复

    2024-03-14 15:16:02       37 阅读
  6. Python: for,while循环语句

    2024-03-14 15:16:02       51 阅读
  7. JDK、JRE和JVM的区别

    2024-03-14 15:16:02       38 阅读
  8. 后端各种格式传参混合vue前端传参

    2024-03-14 15:16:02       44 阅读
  9. FFmpeg--packet数据包和frame数据帧的区别

    2024-03-14 15:16:02       34 阅读
  10. STM32单片机实现串口通信

    2024-03-14 15:16:02       32 阅读
  11. git--协同开发

    2024-03-14 15:16:02       44 阅读
  12. 医药电商大数据实践

    2024-03-14 15:16:02       39 阅读