libtorch 分类C++推断

分类C++推断

#pragma once

#define _CRT_SECURE_NO_WARNINGS
#include<torch/torch.h>
#include<torch/script.h>
#include<iostream>
#include <windows.h>
#include <tuple>
#include <string>
#include <opencv2/opencv.hpp>
#include <opencv2/core/core.hpp>       //核心组件
#include <opencv2/highgui/highgui.hpp> //GUI
#include <opencv2/imgproc/imgproc.hpp> //图像处理
#include <torch/torch.h>
#include <ATen/Parallel.h>
#include<map>

using namespace std;
using namespace cv;
using namespace torch;


class __declspec(dllexport)Runtime_Classification_new
{
   
public:
	Runtime_Classification_new(const string model_path, int num_threads_, int image_size, int image_channle);
	int change_parameters(); //模型参数解析

	/*图片输入进行推断*/
	int Runing_infer(vector<Mat> &orc_image,vector<float> if_score_limit, vector<vector<float>>&result);

	~Runtime_Classification_new();

private:
	string model_path;
	int image_size;
	int image_channle;
	int num_threads_;
};

#include "Runtime_Classification_new.h"



static torch::jit::script::Module model_Classification;
static torch::Device device_all_Classification = torch::kCPU;   //首先设置为CPU版本
static void Stringsplit(string str, const char split, vector<string>&res)
{
   
	istringstream iss(str);	// 输入流 	
	string token;			// 接收缓冲区 	
	while (getline(iss, token, split))	// 以split为分隔符 	
	{
   
		res.push_back(token);
	}
}
/*将一个路径完全分割*/
static vector<string>get_Filepath_many_information(string path)
{
   
	std::vector<std::string> information;
	const char* file;
	file = path.c_str();
	char szDrive[_MAX_DRIVE];   //磁盘名
	char szDir[_MAX_DIR];       //路径名
	char szFname[_MAX_FNAME];   //文件名
	char szExt[_MAX_EXT];       //后缀名
	_splitpath(file, szDrive, szDir, szFname, szExt); //分解路径
	std::string File_name1 = szDrive;
	std::string File_name2 = szDir;
	std::string File_name3 = szFname;
	std::string File_name4 = szExt;
	information.push_back(File_name1);
	information.