//
|
// 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(unsigned char *pBuf, float * pFeature);
|
float ReID_Compare(float *pFeature1, float *pFeature2);
|
void ReID_Release();
|
};
|
|
|
#endif //INC_03_REID_STRONG_BASELINE_REID_FEATURE_H
|
