26 lines
773 B
C
26 lines
773 B
C
|
#ifndef _RETINANET_CLASSIFY_INFERENCE_H_
|
||
|
|
#define _RETINANET_CLASSIFY_INFERENCE_H_
|
||
|
|
|
||
|
|
#include "preprocess.h"
|
||
|
|
#include "inference.h"
|
||
|
|
#include "postprocess.h"
|
||
|
|
|
||
|
|
using namespace nvinfer1;
|
||
|
|
using namespace nvcaffeparser1;
|
||
|
|
using namespace std;
|
||
|
|
|
||
|
|
class RetinanetClassifyInference: public Inference
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
RetinanetClassifyInference();
|
||
|
|
~RetinanetClassifyInference();
|
||
|
|
|
||
|
|
int RetinanetClassifyInferenceInit(ModelInfo* pRetinanetClassifyModelInfo, const std::string& strModelName, const std::string& strDeployName, const std::string& strEngineName);
|
||
|
|
int RetinanetClassifyInferenceDeInit(void);
|
||
|
|
bool RetinanetClassifyInferenceModel(cv::Mat& frame);
|
||
|
|
|
||
|
|
private:
|
||
|
|
ModelInfo* pRetinanetClassifyModelInfo_ = nullptr;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif //END OF _RETINANET_CLASSIFY_INFERENCE_H_
|