Blame view

src/ai_engine_module/VehicleAnalysis.h 580 Bytes
eac85cd5   Hu Chunming   调通va
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
  #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);
  
20396d5c   Hu Chunming   添加车头车尾算法
19
      int detect(vector<sy_img> vec_img);
eac85cd5   Hu Chunming   调通va
20
21
22
23
24
25
26
27
28
29
  
  private:
      void release();
  
  private:
      void* m_handle{nullptr};
      va_param param;	
      
      float m_threshold;
  	int m_max_batchsize{1};
eac85cd5   Hu Chunming   调通va
30
31
32
33
  };
  
  
  #endif      // __AI_ENGINE_MODULE_VEHICLE_ANALYSIS_H__