4.3 C++ 推理示例
本文以目标检测任务为例,介绍如何在 C++ 环境中完成模型推理。目标检测是一项关键的计算机视觉任务,旨在识别图像或视频中各类目标的具体位置和类别。示例使用 YOLOv5 模型,基于 ONNX 推理框架和 SpaceMIT 提供的硬件加速库进行部署与推理。
克隆代码
git clone https://gitee.com/bianbu/spacemit-demo.git ~
构建与运行
cd spacemit_demo/examples/CV/yolov5/cpp
mkdir build
cd build
cmake ..
make -j8
./yolov5_demo --model ../../model/yolov5_n.q.onnx --image ../../data/test.jpg
推理结果默认保存为 result.jpg
,如下图所示: