28 lines
778 B
C++
28 lines
778 B
C++
#ifndef _YOLOV5_CLEAR_DETECT_INFERENCE_H_
|
|
#define _YOLOV5_CLEAR_DETECT_INFERENCE_H_
|
|
|
|
#include "preprocess.h"
|
|
#include "inference.h"
|
|
#include "postprocess.h"
|
|
|
|
using namespace nvinfer1;
|
|
using namespace nvcaffeparser1;
|
|
using namespace std;
|
|
|
|
class YoloV5ClearDetectInference: public Inference
|
|
{
|
|
public:
|
|
YoloV5ClearDetectInference();
|
|
~YoloV5ClearDetectInference();
|
|
|
|
int YoloV5ClearDetectInferenceInit(ClearModelInfo* pYoloV5ClearModelInfo, const std::string& strModelName, const std::string& strEngineName);
|
|
int YoloV5ClearDetectInferenceDeinit(void);
|
|
int YoloV5ClearDetectInferenceModel(cv::Mat& frame, std::vector<ClearDetection>& vecRes);
|
|
|
|
private:
|
|
ClearModelInfo* pYoloV5ClearModelInfo_ = nullptr;
|
|
};
|
|
|
|
|
|
|
|
#endif //END OF _YOLOV5_CLEAR_DETECT_INFERENCE_H_
|