Hu Chunming / vpt_ascend_arm

/*
 * @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};
};