YOLOv6在LabVIEW中的推理部署（含源码）
前言
YOLOv6 是美团视觉智能部研发的一款目标检测框架，致力于工业应用。如何使用python进行该模型的部署，官网已经介绍的很清楚了，但是对于如何在LabVIEW中实现该模型的部署，笔者目前还没有看到相关介绍文章，所以笔者在实现YOLOv6 ONNX 在LabVIEW中的部署推理后，决定和各位读者分享一下如何使用LabVIEW实现YOLOv6的目标检测。
一、什么是YOLOv6
- YOLOV6官方代码：https://github.com/meituan/YOLOv6
YOLOv6 是美团视觉智能部研发的一款目标检测框架，致力于工业应用。本框架同时专注于检测的精度和推理效率，在工业界常用的尺寸模型中：YOLOv6-nano 在 COCO 上精度可达 35.0% AP，在 T4 上推理速度可达 1242 FPS；YOLOv6-s 在 COCO 上精度可达 43.1% AP，在 T4 上推理速度可达 520 FPS。在部署方面，YOLOv6 支持 GPU（TensorRT）、CPU（OPENVINO）、ARM（MNN、TNN、NCNN）等不同平台的部署，极大地简化工程部署时的适配工作。
从模型性能对比图来看，YOLOv6的性能也是再创新高。YOLOv6-s在COCO上精度达到了43.1%AP，在T4上推理速度也高达520FPS！
YOLOv6提供了支持 ONNX、TensorRT、NCNN 和 Openvino 的部署版本。今天我们一起来看一下如何在LabVIEW中部署YOLOv6 ONNX。
二、环境搭建
1、部署本项目时所用环境
- 操作系统：Windows10
- python：3.6及以上
- LabVIEW：2018及以上 64位版本
- AI视觉工具包：techforce_lib_opencv_cpu-1.0.0.73.vip
- onnx工具包：virobotics_lib_onnx_cuda_tensorrt-1.0.0.16.vip【1.0.0.16及以上版本】
2、LabVIEW工具包下载及安装
- AI视觉工具包下载与安装参考：
https://blog.csdn.net/virobotics/article/details/123656523 - onnx工具包下载与安装参考：
https://blog.csdn.net/virobotics/article/details/124998746
三、模型的获取与转化
注意：本教程已经为大家提供了YOLOV6的模型，可跳过本步骤，直接进行步骤四-推理。若是想要了解YOLOV6的onnx模型如何获取，则可继续阅读本部分内容。
下面我们来介绍两种模型获取的方式(以yolov6s为例，想要获取其他模型的方式也一样，只需要修改名字即可)
1、方式一：直接在官网下载yolov6s的onnx模型
具体地址如下:https://github.com/meituan/YOLOv6/tree/main/deploy/ONNX
单击YOLOv6-s，可直接下载yolov6s.onnx
2、方式二：将标准模型pt转化为onnx（较为复杂）
1、下载并安装YOLOv6所依赖的库
在https://github.com/meituan/YOLOv6中下载YOLOv6源码并解压
在YOLOv6-main文件夹中打开cmd，输入以下指令安装需要的库：
pip3 install -U pip && pip3 install -r requirements.txt
注意：YOLOv6比V5多了一个addict库，如果之前已经安装过了YOLOv5的相关库，也可以只下载一个addict
2、安装pycocotools
pip install pycocotools
！！！注：若安装pycocotools时遇到ERROR: Could not build wheels for pycocotools ……，则使用以下指令来安装：
pip install pycocotools-windows
3、从基准表下载标准预训练模型
https://github.com/meituan/YOLOv6
将模型放置到如下路径”\YOLOv6-main\”
4、将标准模型pt转化为onnx
切换到YOLOv6-main文件夹路径下，运行以下指令实现模型的转换
python ./deploy/ONNX/export_onnx.py --weights yolov6s.pt --img 640 --batch 1
输出如下，得到yolov6s.onnx
四、在LabVIEW实现YOLOV6的部署推理
本项目整体的文件结构如下图所示，各位读者可在文章末尾链接处下载整个项目源码。
本例中使用LabVIEW ONNX工具包中的Create_Session.vi载入onnx模型，可选择使用cpu,cuda进行推理加速。
1、查看模型
我们可以使用netron 查看ppyoloe_crn_s_300e_coco.onnx的网络结构，浏览器中输入链接：https://netron.app/，点击Open Model，打开相应的网络模型文件即可。
查看模型属性，可看到模型的输入输出如下图所示:
我们发现，该模型输出和YOLOx输出一直，均为1x8400x85
2、项目运行
将我们已经转化好的onnx模型放置到model文件夹中，打开yolov6_onnx_camera.vi,在前面板中修改程序中加载的模型路径为实际模型路径，本项目中已经将YOLOV6 onnx模型【yolov6s.onnx】放置到了model文件夹中，如需其他模型，读者也可自行放置到model文件及下，实现模型的加载。运行程序，可得到目标检测的结果。
-
主程序源码如下（可以选择使用CUDA或者TensorRT进行推理加速）：
-
运行结果如下：
五、完整项目下载链接
链接：https://blog.csdn.net/virobotics/article/details/126356929?spm=1001.2014.3001.5501
总结
以上就是今天要给大家分享的内容。大家可根据链接下载相关源码与模型。
如果有问题可以在评论区里讨论，提问前请先点赞支持一下博主哦，如您想要探讨更多关于LabVIEW与人工智能技术，欢迎加入我们的技术交流群：705637299。进群请备注：LabVIEW机器学习
**如果文章对你有帮助，欢迎✌关注、👍点赞、✌收藏
