Tracore
Public Member Functions | List of all members
algo::NStage Class Reference

#include <NStage.h>

Public Member Functions

 NStage (size_t max_frame_skip, double penalty_value, size_t max_tracklet_count)
 
void CreateObjectGraph (DirectedGraph &graph, const core::DetectionSequence &detections)
 
void CreateTrackletGraph (DirectedGraph &obj_graph, DirectedGraph &tlt_graph, size_t frame_count)
 
void ExtractTracks (DirectedGraph &tlt_graph, size_t depth, std::vector< core::TrackletPtr > &tracks)
 

Detailed Description

Implementation of the two-staged graph-based multi-object tracker.

Constructor & Destructor Documentation

algo::NStage::NStage ( size_t  max_frame_skip,
double  penalty_value,
size_t  max_tracklet_count 
)

Initializes the algorithm wih the given values.

Parameters
max_frame_skipThe maximum edge length to link objects
penalty_valueThe Edge value to link to source and sink
max_tracklet_countThe maximum number of tracklets to create

Member Function Documentation

void algo::NStage::CreateObjectGraph ( DirectedGraph &  graph,
const core::DetectionSequence detections 
)

Creates a graph with vertices for every detected object

Parameters
graphThe graph to write into
detectionsThe objects to use for the graph
void algo::NStage::CreateTrackletGraph ( DirectedGraph &  obj_graph,
DirectedGraph &  tlt_graph,
size_t  frame_count 
)

Reduces the object graph into linked tracklets.

Parameters
obj_graphThe object graph to reduce
tlt_graphThe graph to write the tracklets in
frame_countThe frame count of the object graph
void algo::NStage::ExtractTracks ( DirectedGraph &  tlt_graph,
size_t  depth,
std::vector< core::TrackletPtr > &  tracks 
)

Extracts the finished tracks from the given tracklet graph.

Parameters
tlt_graphThe tracklet graph to extract from
depthThe depth to flatten the tracklets to
tracksThe vector to write the extracted tracks in

The documentation for this class was generated from the following files: