22 lines
363 B
C++
22 lines
363 B
C++
#pragma once
|
|
|
|
#include <opencv2/core/core.hpp>
|
|
|
|
typedef struct Detection2d {
|
|
int classIdx;
|
|
float confidence;
|
|
cv::Rect bbox;
|
|
}_detection2d;
|
|
|
|
class OnnxDetector
|
|
{
|
|
public:
|
|
static OnnxDetector* CreateInstance();
|
|
|
|
public:
|
|
virtual int loadOnnxModel(const char* filename, cv::Size2f inferSize) = 0;
|
|
virtual std::vector<Detection2d> detect(cv::Mat image) = 0;
|
|
|
|
};
|
|
|