human_gather_statistics.h
1.08 KB
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
/*
* @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 "stl_aiplatform_header.h"
#include "task_param_manager.h"
#include "./include.h"
namespace ai_engine_module
{
namespace human_gather_statistics
{
typedef struct result_data_t
{
long long id;
sy_img *img;
std::vector<box_t> boxes;
std::string task_id;
} result_data_t;
class human_gather_statistics
{
private:
task_param_manager *m_task_param_manager;
map<string, int> taskid_framecount;
static long long gid_;
public:
int human_gather_statistics_init();
std::vector<result_data_t> human_gather_statistics_process(set<string>& taskin_play_id, sy_img* images, vector<onelevel_det_result> &ol_det_result);
std::vector<result_data_t> human_gather_statistics_process2(vector<string>& taskin_play_id, sy_img* images, vector<onelevel_det_result> &ol_det_result);
};
}
}