YOLOv8主要命令讲解

YOLOv8主要有三个常用命令,分别是:train(训练)、predict(预测)、export(转化模型格式),下面我将展开讲讲三个常用命令的常用参数与具体使用方法。

一、训练

通过自己标注的数据集,训练自己所需的模型,训练完成的模型存储在runs文件夹里面。

yolo train data=data/data.yaml model=yolov8s.pt epochs=300 imgsz=640 batch=8 workers=0 device=0

data:训练的参数文件,具体如下:

train: xxx/xxx/images/train    xxx/xxx为训练集图片根目录地址,一定要是绝对路径
val: xxx/xxx/images/val
nc: 2 #训练集的标签名称数量,就是下面names列表参数数量
names: ["1""2"]#标签名称

model:预训练模型,可以是YOLOv8官方的模型,也可以是自己训练的模型,推荐使用yolov8s.pt

epochs:训练轮数,在数据集没问题的情况下,轮数多了,acc(精准度)会提高,loss会降低,但是他会有一个临界值,轮数超过一定数量acc和loss都不在变化,就可以停止了,这个临界值需要凭借经验和数据集数量自己去判断,可以跑个100000轮次看一下到多少停止变化。

imgsz:训练时图片大小,根据自己训练的图片尺寸和要识别的物体大小进行判断,训练的图片尺寸大的建议使用640,图片小的使用320。

batch:训练时每次传入GPU的图片数量,对显卡的显存要求高。数据集train里面的图片数量 / batch的大小即为每轮训练里面的读取图片特征值的轮数,batch越高精度会有所提升,训练也越快。

workes:数据装载时cpu所使用的线程数,可以查看自己的cpu的线程数量进行设置,越高给显卡传输图片速度越快,如果你的显卡性能很强,可以提高workes的值。

device:训练所使用的设备,默认为cpu,0为显卡,如果有多个显卡可以0,1,2…往后排开。

其中主要影响训练速度的有:epochs(轮数越大时间越长)、imgsz(图片尺寸越大时间越长)、batch(越小时间越长)、workes(越小时间越长) 可以打开任务管理器查看CPU和GPU的占用情况调节以上参数

二、预测

yolo predict model=yolov8n.pt source='ultralytics/assets/' device=0

对训练完成的模型,使用测试图片或视频进行测试,测试完成的数据保存在runs文件夹里面。

model:要测试的模型,可以是YOLOv8官方的模型,也可以是自己训练的模型

source:要测试的图片的地址,使用绝对路径和相对路径都可以

device:测试图片使用的设备

三、模型转化

使用该指令将YOLOv8的pt模型转化为onnx模型,可以进一步转化为自己所需的类型。

yolo export model=yolov8n.pt format=onnx

model:要转化的模型

format:要转化成的模型类型

下面是官方写的其他export参数
在这里插入图片描述

相关推荐

最近更新

  1. TCP协议是安全的吗?

    2024-05-02 18:34:02       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-05-02 18:34:02       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-05-02 18:34:02       19 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-05-02 18:34:02       20 阅读

热门阅读

  1. 《21天学通C++》(第十三章) 类型转换运算符

    2024-05-02 18:34:02       10 阅读
  2. MyBatis:mybatis入门

    2024-05-02 18:34:02       11 阅读
  3. MySQL:多表查询和事务

    2024-05-02 18:34:02       15 阅读
  4. 程序员副业可用的四大原则

    2024-05-02 18:34:02       13 阅读
  5. Circuits--Sequential--Finite5

    2024-05-02 18:34:02       11 阅读
  6. BKPUNIX

    2024-05-02 18:34:02       9 阅读
  7. 单片机与Python串口通讯

    2024-05-02 18:34:02       12 阅读
  8. Python Json数据解析

    2024-05-02 18:34:02       13 阅读
  9. 单机多GPU的训练及debug中vscode下launch.json内容设置

    2024-05-02 18:34:02       10 阅读
  10. 【Golang】使用 GORM 的 Scopes 进行查询

    2024-05-02 18:34:02       13 阅读