YOLOV8模型测试、训练及导出onnx模型

一、环境配置

1、下载源码

git clone https://github.com/ultralytics/ultralytics.git

2、安装python包

pip install -r requirements.txt

3、安装ultralytics(源码安装的方式)

# 在ultralytics-main的文件目录里面，运行以下命令
pip install -e .

备注：
如果你使用了官方推荐的pip安装，源码就会在conda环境下
如果使用源码安装的方式，修改代码只需要在自己下载的源码里修改(这种方式更加推荐)。

4、测试ultralytics是否安装成功，运行以下的pytho

import ultralytics
ultralytics.checks()

反馈以下结果，则说明ultralytics安装成功

Ultralytics YOLOv8.0.175 🚀 Python-3.8.18 torch-2.0.1+cu117 CUDA:0 (NVIDIA GeForce RTX 2080 SUPER, 7974MiB)
Setup complete ✅ (16 CPUs, 31.3 GB RAM, 257.3/884.8 GB disk)

二、模型测试

使用V8的官方命令解析

yolo task=detect    mode=train   data=<data.yaml path>      model=yolov8n.pt        args...
          classify                predict        coco-128.yaml                       yolov8n-cls.yaml             args...
          segment               val                                                                         yolov8n-seg.yaml       args...
            pose                                                                                                      yolov8n.pt        format=onnx  args...
           export   

YOLOv8 可以直接在命令行界面 (CLI) 中使用“yolo”命令来执行各种任务和模式，并接受其他参数，即“imgsz=640”。

运行yolov8的目标检测模型

yolo task = predict model = yolov8n.pt source = '/**/1.jpg' , save = False

参数save用来决定是否保存包含预测框的图片
运行yolov8-seg的目标检测与分割模型

yolo task = segment mode = predict model = yolov8n-seg.pt source = '/**/1.jpg'

运行yolov8-cls的分类模型

yolo task = classify mode = predict model = yolov8n-cls.pt source = '/**/1.jpg'

运行yolov8-pose的目标检测与姿态模型

yolo task = pose mode = predict model = yolov8n-pose.pt source = '/**/1.jpg'

三、模型训练

训练yolov8的目标检测模型

yolo detect  train data=./ultralytics/cfg/datasets/coco128.yaml  model = yolov8n.yaml imgsz = 416

备注：

• 修改图片尺寸可以在命名行中指定imgsz，或者在default.yaml中指定
• model可以是预训练模型，也可以yaml文件，其中yolo后面加的n或者s等等来决定采用哪种模型尺寸

训练yolov8的目标检测与姿态模型

yolo pose  train data=./ultralytics-main/ultralytics/cfg/datasets/coco128.yaml  model = yolov8n-pose.yaml

报了以下错误

  File "/home/**/yolov8-python/ultralytics-main/ultralytics/data/dataset.py", line 122, in get_labels
    len_cls, len_boxes, len_segments = (sum(x) for x in zip(*lengths))
ValueError: not enough values to unpack (expected 3, got 0)

这个错误，可能是官方提供的最新代码(时间截至到2023年9月12号)有问题，可以尝试使用旧一点的版本。我在官方源码下载Branches为val_period，训练yolov8的目标检测与姿态模型是没有问题的。

导出onnx模型

yolo export model=/**/weights/best.pt format=onnx opset=12

备注：
在需要训练自己的数据集、更改训练的配置参数或者模型网络结构
修改地方的主要是数据集yaml文件(例如coco128.yaml,或者创建一个自己数据集的yaml文件)、配置参数default.yaml(里面包括很多参数，包括图片尺寸、是否使用预训练权重等等)及模型网络结构yaml文件(例如yolov8.yaml)