YOLOv8进行训练、验证和推理记录

本文记录一下如何使用yolov8进行训练、验证和推理的详细过程。

一、代码下载和环境配置

直接参考我的上一篇文章:Yolov8复现及简单测试记录​​​​​​

二、数据集准备

首先声明一下,yolov8对于数据集的要求只能是voc格式或者yolo格式。

这里提供一个小目标烟头检测的VOC格式数据集小目标烟头检测VOC格式数据集和data.yaml配置文件

1.VOC标注格式数据集的使用

(1)把图片存放在项目下的dataset\VOCdevkit\JPEGImages目录中(没有的话就自己新建一个),图片后缀需要一致,比如都是jpg或者png等等,不支持混合的图片后缀格式,比如一些是jpg,一些是png。

(2)把VOC标注格式的XML文件存放在dataset\VOCdevkit\Annotations目录中。

(3)打开xml2txt.py。在这个文件中其会把Annotations中的XML格式标注文件转换到txt中的yolo格式标注文件。其中xml2txt.py中的postfix参数是JPEGImages的图片后缀,修改成图片的后缀即可,默认为jpg。比如我的图片都是png后缀的,需要把postfix修改为png即可。修改完成后右键Run运行xml2txt.py。

其中运行这个文件的时候,输出信息会输出你的数据集的类别。

(4)修改 data.yaml文件中的参数。你需要把运行xml2txt.py文件后输出的类别列表复制到data.yaml中的names中,并且修改nc为你的类别数,也就是names中类别个数。

(5)运行split_data.py,这个文件是划分训练、验证、测试集。其中支持修改val_size验证集比例和test_size测试集比例,可以在split_data.py中找到对应的参数进行修改,然后postfix参数也是你的图片数据集后缀格式,默认为jpg,如果你的图片后缀不是jpg结尾的话,需要修改一下这个参数。

split_data.py文件在运行的时候不会有任何显示,如何确定它是否运行完成就是看dataset文件夹中是否出现了images和lables两个文件夹。

2.YOLO标注格式数据集的使用

(1)把图片存放在dataset\VOCdevkit\JPEGImages中,图片后缀需要一致,比如都是jpg或者png等等,不支持混合的图片后缀格式,比如一些是jpg,一些是png。

(2)把YOLO标注格式的TXT文件存放在dataset\VOCdevkit\txt中。

(3)运行split_data.py,这个文件是划分训练、验证、测试集。其中支持修改val_size验证集比例和test_size测试集比例,可以在split_data.py中找到对应的参数进行修改,然后postfix参数也是你的图片数据集后缀格式ÿ

猜你喜欢

转载自blog.csdn.net/m0_63604019/article/details/130091686
今日推荐