Update YOLOv8-ONNXRuntime-CPP (#3455)
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
1c4a363497
commit
69a2d70a78
5 changed files with 453 additions and 0 deletions
83
examples/YOLOv8-ONNXRuntime-CPP/inference.h
Normal file
83
examples/YOLOv8-ONNXRuntime-CPP/inference.h
Normal file
|
|
@ -0,0 +1,83 @@
|
|||
#pragma once
|
||||
|
||||
#define _CRT_SECURE_NO_WARNINGS
|
||||
#define RET_OK nullptr
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <stdio.h>
|
||||
#include "io.h"
|
||||
#include "direct.h"
|
||||
#include "opencv.hpp"
|
||||
#include <Windows.h>
|
||||
#include "onnxruntime_cxx_api.h"
|
||||
|
||||
|
||||
enum MODEL_TYPE
|
||||
{
|
||||
//FLOAT32 MODEL
|
||||
YOLO_ORIGIN_V5 = 0,
|
||||
YOLO_ORIGIN_V8 = 1,//only support v8 detector currently
|
||||
YOLO_POSE_V8 = 2,
|
||||
YOLO_CLS_V8 = 3
|
||||
};
|
||||
|
||||
|
||||
typedef struct _DCSP_INIT_PARAM
|
||||
{
|
||||
std::string ModelPath;
|
||||
MODEL_TYPE ModelType = YOLO_ORIGIN_V8;
|
||||
std::vector<int> imgSize={640, 640};
|
||||
|
||||
int classesNum=80;
|
||||
float RectConfidenceThreshold = 0.6;
|
||||
float iouThreshold = 0.5;
|
||||
bool CudaEnable = false;
|
||||
int LogSeverityLevel = 3;
|
||||
int IntraOpNumThreads = 1;
|
||||
}DCSP_INIT_PARAM;
|
||||
|
||||
|
||||
typedef struct _DCSP_RESULT
|
||||
{
|
||||
int classId;
|
||||
float confidence;
|
||||
cv::Rect box;
|
||||
}DCSP_RESULT;
|
||||
|
||||
|
||||
class DCSP_CORE
|
||||
{
|
||||
public:
|
||||
DCSP_CORE();
|
||||
~DCSP_CORE();
|
||||
|
||||
public:
|
||||
char* CreateSession(DCSP_INIT_PARAM &iParams);
|
||||
|
||||
|
||||
char* RunSession(cv::Mat &iImg, std::vector<DCSP_RESULT>& oResult);
|
||||
|
||||
|
||||
char* WarmUpSession();
|
||||
|
||||
|
||||
template<typename N>
|
||||
char* TensorProcess(clock_t& starttime_1, cv::Mat& iImg, N& blob, std::vector<int64_t>& inputNodeDims, std::vector<DCSP_RESULT>& oResult);
|
||||
|
||||
|
||||
private:
|
||||
Ort::Env env;
|
||||
Ort::Session* session;
|
||||
bool cudaEnable;
|
||||
Ort::RunOptions options;
|
||||
std::vector<const char*> inputNodeNames;
|
||||
std::vector<const char*> outputNodeNames;
|
||||
|
||||
|
||||
int classesNum;
|
||||
MODEL_TYPE modelType;
|
||||
std::vector<int> imgSize;
|
||||
float rectConfidenceThreshold;
|
||||
float iouThreshold;
|
||||
};
|
||||
Loading…
Add table
Add a link
Reference in a new issue