20 lines
680 B
C++
20 lines
680 B
C++
#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();
|
|
};
|
|
}
|