本文从一个小白视角出发,尽可能帮助刚接触机器视觉并想训练模型的同学们。
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
视频下方也有给出模型下载地址,我这里也给出吧: