[深度学习]基于C++和onnxruntime部署yolov13的onnx模型

云未归来

发布于 2025-07-22 15:24:12

40000

代码可运行

运行总次数：0

代码可运行

基于C++和ONNX Runtime部署YOLOv13的ONNX模型，可以遵循以下步骤：

准备环境：首先，确保已经下载后指定版本opencv和onnruntime的C++库。
模型转换： 安装好yolov13环境并将YOLOv13模型转换为ONNX格式。这通常涉及使用深度学习框架（如PyTorch或TensorFlow）加载原始模型，并导出为ONNX格式。转换指令

# End-to-End ONNX
yolo export model=yolov13{n/s/m/b/l/x}.pt format=onnx opset=13 simplify

C++环境配置：在CMakeLists.txt项目中正确引用了opencv和ONNX Runtime的头文件，并链接到相应的库。这允许在C++代码中使用ONNX Runtime的功能。
加载模型：使用ONNX Runtime的API加载转换后的YOLOv13 ONNX模型。
执行推理：通过ONNX Runtime的推理引擎，将图像数据输入到模型中，并执行目标检测任务。
处理结果：解析模型输出的结果，这通常涉及将输出的张量数据转换为可理解的检测结果，如边界框坐标和类别标签。

通过这些步骤，可以在C++环境中利用ONNX Runtime高效地部署YOLOv13模型，实现实时的目标检测功能。

【测试环境】

windows10 x64 vs2019 cmake==3.30.1 onnxruntime==1.16.3 opencv==4.9.0 【使用步骤】首先cmake生成exe文件，然后将onnxruntime.dll和onnxruntime_providers_shared.dll放到exe一起，不然会提示报错0xc000007b，这是因为系统目录也有个onnxruntime.dll引发冲突，并把car.mp4也放到exe一起。运行直接输入 yolov13.exe 注意onnx路径要是你真实路径我的onnx路径是我桌面上地址

【代码调用】

注意onnxruntime使用的cpu版本库，如需使用GPU还需要修改代码才行

#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgcodecs.hpp>
#include <iostream>
#include <string>
#include "YOLO13.hpp"

int main() {
    // Paths to the model, labels, test image, and save directory
    const std::string labelsPath = "../models/coco.names";
    const std::string imagePath = "../data/dog.jpg";           // Image path
    const std::string savePath = "../data/dog_detections.jpg";   // Save directory

    // Model path for YOLOv13
    const std::string modelPath = "../models/yolov13n.onnx";   // YOLOv13

    // Initialize the YOLO detector with the chosen model and labels
    bool isGPU = true; // Set to false for CPU processing
    YOLO13Detector detector(modelPath, labelsPath, isGPU);

    // Load an image
    cv::Mat image = cv::imread(imagePath);
    if (image.empty()) {
        std::cerr << "Error: Could not open or find the image!\n";
        return -1;
    }

    // Detect objects in the image and measure execution time
    auto start = std::chrono::high_resolution_clock::now();
    std::vector<Detection> results = detector.detect(image);
    auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(
                        std::chrono::high_resolution_clock::now() - start);

    std::cout << "Detection completed in: " << duration.count() << " ms" << std::endl;

    // Draw bounding boxes on the image
    detector.drawBoundingBox(image, results); // Simple bounding box drawing
    // detector.drawBoundingBoxMask(image, results); // Uncomment for mask drawing

    // Save the processed image to the specified directory
    if (cv::imwrite(savePath, image)) {
        std::cout << "Processed image saved successfully at: " << savePath << std::endl;
    } else {
        std::cerr << "Error: Could not save the processed image to: " << savePath << std::endl;
    }

    // Display the image
    cv::imshow("Detections", image);
    cv::waitKey(0); // Wait for a key press to close the window

    return 0;
}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2025-06-27，如有侵权请联系 cloudcommunity@tencent.com 删除

c++