移动网站如何优化排名网站建设图文片
- 🍨 本文为🔗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