#ifndef _sdk_reid_h_
#define _sdk_reid_h_

#include <torch/script.h>

class reid
{
private:
    bool is_gpu_;
    bool start_;
    torch::jit::script::Module module_;
public:
    explicit reid(const int gpu_index, const char *module_path);
    ~reid();

private: 
    int init(const int gpu_index, const char *module_path);
    void release();
public: 
    int extract(unsigned char *img, const int w, const int h, const int c, float *feature);
    float compare(const float *feat1, const float *feat2);
    int feat_size(){return 2048;}
};


#endif