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

1 前言

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

pytorch版本1.7.1       OpenCV 4.5.4

2 代码

#include
#include
#include
#include 
#include
#include
#include
#include
using namespace cv;
using namespace std;

class ONNXClassifier
{
public:
	ONNXClassifier(const std::string& model_path, const std::string& label_path, cv::Size _input_size);
	void Classify(const cv::Mat& input_image, std::string& out_name, double&