Pytorch学习系列之十二：OpenCV（C++）加载onnx分类模型进行推理

thequitesunshine007

已于 2022-12-19 15:45:11 修改

阅读量6.4k

点赞数

CC 4.0 BY-SA版权

分类专栏： Pytorch学习文章标签： pytorch

于 2022-04-05 23:12:59 首次发布

本文链接：https://blog.csdn.net/thequitesunshine007/article/details/123979877

Pytorch学习专栏收录该内容

29 篇文章 ¥49.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何在C++中利用OpenCV 4.5.4加载ONNX模型，该模型由Pytorch 1.7.1训练得到，用于进行图像分类推理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 前言

博客Pytorch学习系列之九：OpenCV加载onnx模型进行推理_thequitesunshine007的博客-CSDN博客_opencv 加载onnx记载了python版本opencv加载onnx模型进行推理的方法。这里记录的是C++版本OpenCV加载onnx模型进行推理。

pytorch版本1.7.1 OpenCV 4.5.4

2 代码

#include<opencv2/highgui/highgui.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include<opencv2/dnn/dnn.hpp>
#include <opencv2/core/utils/logger.hpp>
#include<string>
#include<iostream>
#include<fstream>
#include<vector>
using namespace cv;
using namespace std;

class ONNXClassifier
{
publ

了解本专栏