wangrong
2025-01-23 02819b5c047bb354b0ef2374e7c6a6ac4bbd5bba
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#pragma once
#include "vp_track_node.h"
 
namespace vp_nodes {
    // track node using deep sort
    class vp_dsort_track_node: public vp_track_node
    {
    private:
        /* config data for deep sort algo*/
    protected:
        // fill track_ids using deep sort algo
        virtual void track(int channel_index, const std::vector<vp_objects::vp_rect>& target_rects, 
                        const std::vector<std::vector<float>>& target_embeddings, 
                        std::vector<int>& track_ids) override;
    public:
        vp_dsort_track_node(std::string node_name, vp_track_for track_for = vp_track_for::NORMAL);
        virtual ~vp_dsort_track_node();
    };
}