qq_56529216的博客

原创 YOLOv8训练自己的数据集（包括环境搭建、超参数调整、模型训练、推理、模型导出等）

images存放图片，lables存放txt格式的标签，未划分前将所有图片和标签都保存在train文件夹中，然后运行代码划分数据，我这里是8：2，想调整直接看代码最后一个小数。xml文件存储位置，生成txt文件存储位置，还有你的数据集类别名称，要修改成自己的类别。RUN就开始训练了，最后一行可以设置不同的超参数，更多的超参数可以在default.yaml中设置。网上大多公开数据集标签是xml格式的，这里附上xml标签转换为txt格式的代码。自己根据自己的数据集标签，配置一个yaml文件。