Blame view

src/ai_engine_module/VehicleAnalysisDevice.h 2.75 KB
5a4c4dc0   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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
  #ifndef __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_DEVICE_H__
  #define __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_DEVICE_H__
  
  #include "vehicle_analysis.h"
  
  /*************************************************************************
  * function: va_init
  * purpose: init resources
  * param:
  [in] handle 		- handle
  [in] param 			- init param
  * return: success(0) or error code(<0)
  * notes: null
  *************************************************************************/
  int va_init(void ** handle, va_param param);
  
  /*************************************************************************
  * function: va_release
  * purpose: release sources
  * param:
  [in] handle				- handle
  * return: null
  * notes: null
  *************************************************************************/
  void va_release(void ** handle);
  
  /*************************************************************************
  * function: va_get_version
  * purpose: get sdk version
  * param: null
  * return: null
  * notes: null
  *************************************************************************/
  const char * va_get_version();
  
  /*************************************************************************
  * function: va_batch
  * purpose: vehicle analysis batch
  * param:
  [in] handle				- handle
  [in] img_data_array		- data array
  [in] format				- data format
  [in] batch_size			- batch size
  [in] result				- vehicle analysis result
  * return: success(0) or error code(<0)
  * notes: null
  *************************************************************************/
  int va_batch(void * handle, sy_img * img_data_array, int batch_size, va_result *result);	
  //img_data_arraydevice数据,batch_size==16时调用固定16模型,小于16时调用固定1模型。
  int va_batch16_device(void * handle, sy_img * img_data_array, int batch_size, va_result *result);		
  
  
  
  /*************************************************************************
  * FUNCTION: VA_ComputeSimilarity
  * PURPOSE: 比对相似度(通用)
  * PARAM:
  [in] Afea					- 第一张图片特征
  [in] Bfea					- 第二张图片特征
  [in] featuresize			- 特征长度
  * RETURN:	相似度
  * NOTES:	
  *************************************************************************/
  float va_compute_similarity(float *Afea, float *Bfea, int featuresize);
  
  /*************************************************************************
  * FUNCTION: va_compute_similarity_byvafeature
  * PURPOSE: 比对相似度(针对va特征)
  * PARAM:
  [in] Afea					- 第一张图片va特征
  [in] Bfea					- 第二张图片va特征
  [in] featuresize			- 特征长度VA_FEATURESIZE
  * RETURN:	相似度
  * NOTES:	
  *************************************************************************/
  float va_compute_similarity_byvafeature(void * handle , float *Afea, float *Bfea, int featuresize);
  
  
  #endif  // __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_DEVICE_H__