Blame view

src/ai_engine_module/GatherDetect.h 785 Bytes
f1eb769d   Hu Chunming   1. 添加GatherDetect
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
  /*
   * @Author: yangzilong
   * @Date: 2021-12-10 15:48:49
   * @Last Modified by: yangzilong
   * @Last Modified time: Do not edit
   * @Email: yangzilong@objecteye.com
   * @Description:
   */
  #pragma once
  
  #include <set>
  #include <string>
  #include <vector>
  #include "../ai_platform/task_param_manager.h"
  #include "./ai_engine_header.h"
  #include "../decoder/interface/DeviceMemory.hpp"
  
  struct GatherResult
  {
      long long id;
      vpc_img_info origin_img;
      std::vector<box_t> boxes;
      std::string task_id;
  } ;
  
  class GatherDetect
  {
  public:
      GatherDetect();
  
      std::vector<GatherResult> process(vector<DeviceMemory*> vec_vptMem, vector<onelevel_det_result> &ol_det_result, algorithm_type_t type);
  
  private:
      task_param_manager *m_task_param_manager{nullptr};
      long long gid_{0};
  };