YOLOv8上手——随手录

发布时间 2023-11-05 10:34:06作者: waterdoor

YOLO8

1.配置环境pytorch环境以及cudn环境就不详解,只需要根据pytorch官网安装后再pip opencv即可正常运行
2.git yolov8(ps: 现在也可以直接通过pip下载yolo8了)

运行起来

可以看官网给出的文档,里面有许多案例以及参数的含义,一下内容都是根据博主个人常用的,不够详细敬请谅解

from ultralytics import YOLO
import os
import torch
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
model = YOLO('yolov8n.pt')
model.train(data="coco128.yaml", epochs=3, workers=0)

ps:本案例是基于windows环境运行的

通过上述代码,即可完成简单的yolo8对coco128张数据进行3次简单的训练测试,完成即表明配置与yolov8都可以正确使用了。

# 进行图片测试
model.predict(data="./assets/img", save=True)
# 视频追踪
model.track("./test.mp4", show=True)

save=True是保存图片到yolov8设置的环境中,在输出中可以查看,show是在运行时展示图片
./指的是当前目录的文件,上一级目录用的是../,上上级目录是../../

代码 and 配置

训练自己的数据

通过在cfg中配置的datasets里面查看文件的配置以及修改成自己的数据
image
train: 训练数据-设置自己的数据集目录
val:验证集数据-设置自己的验证集目录

names: 模型yolov8的配置文件,换成自己的labels

然后cfg的model中选择自己要的模型大小和尺寸

以上配置都完成后,按上面的实例运行代码去训练自己的数据集

文件配置

yolov8默认是用的在cfg目录下的default.yaml对数据的超参数进行定义,如果想修改超参数,那么可以自己定义一个新的yaml文件来对超参数进行调参训练
通过

model.train(data="coco128.yaml", epochs=3, workers=0, cfg='./ultralytics/cfg/my_default.yaml')

来加载自定义的超参数

未完成,持续更新!~~