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中找到对应的参数进行修改，