Blame view

src/ai_engine_module/RoadSegAnalysis.h 553 Bytes
2ae58093   Hu Chunming   添加road_seg算法
1
  #include "include.h"
06e0182f   Hu Chunming   修复src图片alignWidth...
2
3
4
5
6
7
8
9
10
11
12
  #include "../village_inc.h"
  #include "road_seg.h"
  
  #include <vector>
  
  using namespace std;
  
  struct RoadInfo{
      std::vector<LineInfo> vec_line;
      std::vector<SegInfo> vec_road;
  };
2ae58093   Hu Chunming   添加road_seg算法
13
14
15
16
17
18
19
20
21
  
  class RoadSegAnalysis
  {
  public:
      RoadSegAnalysis(/* args */);
      ~RoadSegAnalysis();
  
      int init(int devId);
  
06e0182f   Hu Chunming   修复src图片alignWidth...
22
      std::vector<RoadInfo> detect(vector<sy_img> vec_img);
2ae58093   Hu Chunming   添加road_seg算法
23
24
25
26
  
  private:
      int release();
  
06e0182f   Hu Chunming   修复src图片alignWidth...
27
28
      std::vector<SegInfo> parse_seg(rs_result one_result, sy_img src);
  
2ae58093   Hu Chunming   添加road_seg算法
29
30
31
32
  private:
      void* m_handle{nullptr};
      aclrtContext ctx{nullptr};
  };