使用YOLOV8 CLI训练自己的数据集

YOLOV8现在可以直接通过命令行工具运行训练, 推理过程了, 简直傻瓜都能搞两下, 方法如下, 首先在新的conda环境安装ultralytics的包:

pip install ultralytics

接着尝试使用yolov8n来简单做个推理:

yolo task=detect mode=predict model=yolov8n.pt conf=0.25 source=some_picture.jpeg

OK, 能识别出人就行.

接下来我们使用一个安全防护, 包括安全帽与反光衣的数据集, 利用3060显卡, 训练看看.
首先我们把数据集连同标注上传到 https://app.roboflow.com/
具体方法如下:
https://blog.roboflow.com/how-to-train-yolov8-on-a-custom-dataset/

基本上拖进去, 点两下鼠标就可以了.

点击Custom Train and Upload

选择YOLOv8, 点击Get Snippet

系统会帮你打个包, 在miniconda里面, 使用curl下载这个压缩包, 然后如果是windows系统, 就直接用系统的解压缩工具解压.

打开这个reboflow文件夹里面有个data.yaml, 打开, 编辑一下类别名称

3060显卡只有8G显存, 尝试使用batchsize为16来训练, 后面根据情况调整.

另外, 需要保证这torch的这三个, 是使用的gpu版本:

开始训练吧:

yolo task=detect mode=train model=yolov8s.pt data=C:\Users\zunly\OneDrive\ai\yolov8_playground\roboflow\data.yaml epochs=100 imgsz=640 batch=16

最终训练了161轮, 但是第161轮其实就没啥进步空间了:

ok, 接下来就利用这个模型来做个推理, source就是我的自拍视频:

标注好的数据集:
链接：https://pan.baidu.com/s/1GhK1QTGJtCDFoOhAuati0Q
提取码：2d1s