eac85cd5
 Hu Chunming
调通va
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
#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__
|
