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