Blame view

src/ai_engine_module/MotorRainshedAnalysis.h 420 Bytes
f171c20a   Hu Chunming   添加moter_rainshed ...
1
2
  #include "include.h"
  
6f9dffde   Hu Chunming   返回prob
3
  struct MotorRainshedResult {
b1547dfc   Hu Chunming   设置初始值,避免异常值影响
4
5
      int rainshed{1};
      float prob{0.0};
6f9dffde   Hu Chunming   返回prob
6
7
  };
  
f171c20a   Hu Chunming   添加moter_rainshed ...
8
9
10
11
12
13
  class MotorRainshedAnalysis
  {
  public:
      MotorRainshedAnalysis(/* args */);
      ~MotorRainshedAnalysis();
  
581a68a4   Hu Chunming   修正parse_road无返回值导...
14
      int init(int devId, std::string sdk_root);
f171c20a   Hu Chunming   添加moter_rainshed ...
15
  
6f9dffde   Hu Chunming   返回prob
16
      vector<MotorRainshedResult> detect(vector<sy_img> vec_img);
f171c20a   Hu Chunming   添加moter_rainshed ...
17
18
19
20
21
22
23
24
  
  private:
      int release();
  
  private:
      void* m_handle{nullptr};
      aclrtContext ctx{nullptr};
  };