Hu Chunming / village_ascend_arm

#ifndef __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_H__
#define __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_H__

#include "include.h"

#include "vehicle_analysis.h"
#include "vehicle_result.h"

class VehicleAnalysis
{
private:
    /* data */
public:
    VehicleAnalysis(/* args */);
    ~VehicleAnalysis();

    int init(int devId, int max_batch_size);

    int detect(vector<ImageData> vec_img);

private:
    void release();

private:
    void* m_handle{nullptr};
    va_param param;	
    
    float m_threshold;
	int m_max_batchsize{1};

    aclrtContext ctx{nullptr};
};


#endif      // __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_H__