first commit
This commit is contained in:
19
nodes/track/vp_dsort_track_node.h
Normal file
19
nodes/track/vp_dsort_track_node.h
Normal file
@@ -0,0 +1,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();
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user