【YOLOv8】实战一：手把手教你使用YOLOv8实现实时目标检测

前言

从2015 年首次发布以来，You Only Look Once (YOLO) 系列计算机视觉模型一直是该领域最受欢迎的模型之一。其中，YOLO 架构的核心创新是将目标检测任务视为回归问题，从而使模型同时对所有目标边界框和类别概率生成预测。在过去的八年中，这种架构创新催生了一系列 YOLO 模型。之前也给大家介绍了一些YOLO 模型在LabVIEW上的部署。感兴趣的话可以查看专栏【深度学习：物体识别（目标检测）】本文主要想和各位读者分享YOLOv8在LabVIEW中的部署。

一、YOLOv8简介
YOLOv8 是由 YOLOv5 的发布者 Ultralytics 发布的最新版本的 YOLO。它可用于对象检测、分割、分类任务以及大型数据集的学习，并且可以在包括 CPU 和 GPU 在内的各种硬件上执行。

YOLOv8是一种尖端的、最先进的 (SOTA) 模型，它建立在以前成功的 YOLO 版本的基础上，并引入了新的功能和改进，以进一步提高性能和灵活性。YOLOv8 旨在快速、准确且易于使用，这也使其成为对象检测、图像分割和图像分类任务的绝佳选择。具体创新包括一个新的骨干网络、一个新的 Ancher-Free 检测头和一个新的损失函数，还支持YOLO以往版本，方便不同版本切换和性能对比。

YOLOv8 有 5 个不同模型大小的预训练模型：n、s、m、l 和 x。关注下面的参数个数和COCO mAP（准确率），可以看到准确率比YOLOv5有了很大的提升。特别是 l 和 x，它们是大模型尺寸，在减少参数数量的同时提高了精度。

每个模型的准确率如下

YOLOv8官方开源地址：https://github.com/ultralytics/ultralyticshttps://github.com/ultralytics/ultralytics

二、环境搭建
2.1 部署本项目时所用环境
操作系统：Windows10
python：3.6及以上
LabVIEW：2018及以上 64位版本
AI视觉工具包：techforce_lib_opencv_cpu-1.0.0.98.vip
onnx工具包：virobotics_lib_onnx_cuda_tensorrt-1.0.0.16.vip【1.0.0.16及以上版本】或virobotics_lib_onnx_cpu-1.13.1.2.vip
2.2 LabVIEW工具包下载及安装
AI视觉工具包下载与安装参考：
https://blog.csdn.net/virobotics/article/details/123656523
onnx工具包下载与安装参考：
https://blog.csdn.net/virobotics/article/details/124998746

三、yolov8导出onnx
注意：本教程已经为大家提供了YOLOv8的onnx模型，可跳过本步骤，直接进行步骤四-项目实战。若是想要了解YOLOv8的onnx模型如何导出，则可继续阅读本部分内容。下面我们来介绍onnx模型的导出(以yolov8s为例，想要导出其他模型的方式也一样，只需要修改名字即可)

3.1 安装YOLOv8
YOLOv8的安装有两种方式，pip安装和GitHub安装。

pip安装

pip install ultralytics -i https://pypi.douban.com/simple/

GitHub安装

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

cd ultralytics

pip install -e '.[dev]'

安装完成后就可以通过yolo命令在命令行进行使用了。

3.2 下载模型权重文件
在官方网站中下载我们所需要的权重文件

注意：这一步其实可以不做，我们在第三步导出模型为onnx的时候如果权重文件不存在，会自动帮我们下载一个权重文件，但速度会较慢，所以个人还是建议先在官网中下载权重文件，再导出为onnx模型

3.3 导出模型为onnx
新建一个文件夹名字为“yolov8_onnx”，将刚刚下载的权重文件“yolov8s.pt”放到该文件夹下的models文件夹里

在models文件夹下打开cmd，在cmd中输入以下命令将模型直接导出为onnx模型：

yolo export model=yolov8s.pt format=onnx opset=12

YOLOv8的3个检测头一共有80x80+40x40+20x20=8400个输出单元格，每个单元格包含x,y,w,h这4项再加80个类别的置信度总共84列内容，所以通过上面命令导出的onnx模型的输出维度为1x84x8400。

如果觉得上面方式不方便，那我们也可以写一个python脚本，快速导出yolov8的onnx模型，程序如下：

from ultralytics import YOLO
 
# Load a model
model = YOLO("\models\yolov8s.pt")  # load an official model
 
# Export the model
model.export(format="onnx")

四、项目实践
实现效果：LabVIEW中部署yolov8实现图片推理和视频推理

整个项目工程如下，本次项目以yolov8s为例

model:yolov8模型文件
subvi：子vi
export.py:将yolov8的ptd导出为onnx
yolov8_camera:yolov8摄像头视频实时推理
yolov8_img:yolov8图片推理
准备工作

放置一张待检测图片和coco.name到yolov8_onnx文件夹下，本项目中放置了一张dog.jpg图片；
确保models文件中已经放置了yolov8的onnx模型：yolov8s.onnx；
4.1 YOLOv8在LabVIREW中实现图片推理
1.获取待检测图片、模型文件及类别标签文件路径；

2.模型初始化：加载onnx模型并读取该模型输入和输出的shape；