//
// Created by Scheaven on 2020/4/10.
//

#ifndef INC_01_CPP_SORT_FEATUREENCODER_H
#define INC_01_CPP_SORT_FEATUREENCODER_H

#include "model.h"
#include "../config.h"
#include "../utils/config_util.h"
#include "../utils/nvidia_gpu_util.h"

#define FEATURE_DIM 128
typedef unsigned char uint8;


class FeatureEncoder
{
public:
    static FeatureEncoder* getInstance();
    bool getRectsFeature(const cv::Mat& img, DETECTIONS& detections);
    ~FeatureEncoder();
    void release();

private:
    tensorflow::Session* session;
    static FeatureEncoder* instance;

    FeatureEncoder();

    void mats_buffer(const std::vector<cv::Mat> &imgs, uint8 *buf);
    std::vector<tensorflow::Tensor> output_tensors;
public:
    void test();
};

#endif //INC_01_CPP_SORT_FEATUREENCODER_H