#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();
|
};
|
}
|