Yolov8复现踩坑记录

本文从一个小白视角出发,尽可能帮助刚接触机器视觉并想训练模型的同学们。

1.首先就是针对yolov8的数据集的结构部分

        v8与其他模型有些不同,这里给出数据集结构的大致图:

其中datasets文件需要大家在ultralytics-main文件根目录下自行创建,注意images和labels文件夹中的文件名要一一对应,对于test文件根据你数据集来定,没有也没关系。其次images/train文件中存放的是你数据集图像,labels/train文件中存放对应images/train中图像的.txt文件。

注意data.yaml文件内容也要进行相应的修改,举个栗子:(假如已经cd进入ultralytics-main文件中)

path:datasets/(数据集名字)

train:images/train

valid:images/valid

test:images/test

#classes

name:

 0: (标签名)

2.正式训练时报错:“Dataset 'data.yaml' images not found”

这里的原因是settings.yaml文件中”datasets_dir”路劲设置问题,这里可以在终端输入yolo settings查看文件参数信息yolo settings datasets_dir='ultralytics-main'即可(假如已经cd进入ultralytics-main文件中)

若还有其他需要更改,具体参考下文:

yolov8训练自定义数据集——settings.yaml与自定义数据集.yaml指向路径的配置问题_yolo训练自己的数据集的yaml文件-CSDN博客

3.RuntimeError: There were no tensor arguments to this function

这个问题最初对于小白来说无从下手,github上下载的yolov8源码做了更新,更原先的可能有些不一样,本人也是尝试修改文件的相关参数也无济于事,毕竟也是刚入门,只求能先把模型跑通就行。所以这里给出一个本简单除暴的方法:更换模型。

我这里使用的是b站的yolov8源码,同时该视频也包含了如何跑通模型的一些操作,非常好用!!!!链接如下:包会!YOLOv8训练自己的数据集_哔哩哔哩_bilibili

视频下方也有给出模型下载地址,我这里也给出吧:

GitHub - shuai-cao/YOLOv8_cs

  • 5
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值