VehicleFeature.h 810 Bytes


#ifndef _VEHICLEFEATURES_H_
#define _VEHICLEFEATURES_H_
#include "vehicle_fea_int8.h"
#include "vpd.h"

#ifndef __INT8__
#define __INT8__
typedef unsigned char int8;
#endif

class VehicleFeature {
public:
    VehicleFeature(/* args */);
    ~VehicleFeature();

    int init_vpd(char*dbpath, int gpuid, char* auth_license);
    int process_vpd(sy_img * batch_img, int batch_size, vpd2_result*& result);
    int release_vpd();

    int init_vf(char*dbpath, int gpuid, char* auth_license);
    int process_vf( sy_img * batch_img, int batch_size, int8**& result);
    int release_vf();

private:
    void RectboundCheck(int Width, int Height, int &left, int &top, int &width, int &height);
    
private:
    void* vpd_handle {nullptr};
    void* vf_handle {nullptr};
};

#endif