#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_