第Y7周：训练自己的数据

• 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
• 🍖 原作者：K同学啊

一、下载YOLOv8
官网地址：YOLOv8开源地址

二、配置环境
在命令行输入：pip install ultralytics
官方把代码都整合到了ultralytics包里面，只需安装这一个库就可以了，里面封装了环境所需库。

三、准备工作
目录结构：

Annotations文件夹存放.xml格式的文件，如下所示

images文件夹存放.png格式的文件，如下所示

运行 split train_val.py文件后将得到 train.txt、val.txt、test.txt和 trainval.txt 四个文件，结果如下:

再创建一个voc_label.py文件

运行voc label.py文件生成 train.txt、test.txt、val.txt 文件:

创建一个ab.yaml文件，内容如下：

train: ./train.txt
val: ./val.txtnc: 4
names: ["banana", "snake fruit", "dragon fruit", "pineapple"]

四、运行
在此文件夹下打开cmd，激活环境后在cmd中输入：

yolo task=detect mode =train model=yolov8s.yaml data=E:\YOLO\ultralytics-main\paper_data\ab.yaml epochs=100 batch=24