前言:YOLO系列的目标检测算法已经进展到YOLOv8,但大多是Pytorch框架下实现的Python代码,而我经常会遇到一些C++推理、C++部署的需求,这里以Github上找到的C++代码为例,讲解一下,如何进行YOLOv5的C++推理(CPU或GPU)。需要尝试的同学,请全部按照这个教程来,尤其是软件或压缩包的版本。
1 代码下载
itsnine/yolov5-onnxruntime: YOLOv5 ONNX Runtime C++ inference code. (github.com)
2 环境搭建
无论是CPU环境还是GPU环境,都需要下载并安装VisualStudio2017或者2019、Cmake、OpenCV。此外,还需要下载“onnxruntime”,这个有GPU版本和CPU版本。
1、VS2019安装:安装时,注意要选择“使用C++的桌面开发”以及将安装位置选到系统盘之外。
2、OpenCV安装:OpenCV安装教程_安装opencv-CSDN博客
3、Cmake安装: