//
// Created by Scheaven on 2020/1/3.
//

#ifndef INC_03_REID_STRONG_BASELINE_REID_FEATURE_H
#define INC_03_REID_STRONG_BASELINE_REID_FEATURE_H
#include <torch/script.h>
#include <opencv2/opencv.hpp>
#include <fstream>
#include <string>
#include <iomanip>
#include <stdlib.h>
#include <vector>
#include <iostream>


struct ReID_Feature {
private:
    bool is_gpu;
    //    auto feat;
    torch::jit::script::Module module;
public:
    bool ReID_init(int gpu_id);
    int ReID_size();
//    static unsigned char * extractor(unsigned char *pBuf, unsigned char *pFeature);
    bool ReID_extractor(float *pBuf, float * pFeature);
    float ReID_Compare(float *pFeature1, float *pFeature2);
    void ReID_Release();
};


#endif //INC_03_REID_STRONG_BASELINE_REID_FEATURE_H