OpenShot Library | libopenshot  0.5.0
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
SortTracker Class Reference

#include <sort_filter/sort.hpp>

Public Member Functions

double GetCentroidsDistance (cv::Rect_< float > bb_test, cv::Rect_< float > bb_gt)
 
 SortTracker (int max_age=50, int min_hits=5, int max_missed=7, double min_iou=0.1, double nms_iou_thresh=0.5, double min_conf=0.3)
 
void update (std::vector< cv::Rect > detection, int frame_count, double image_diagonal, std::vector< float > confidences, std::vector< int > classIds)
 

Static Public Member Functions

static double GetIOU (cv::Rect_< float > bb_test, cv::Rect_< float > bb_gt)
 

Public Attributes

int _max_age
 
int _max_missed
 
double _min_conf
 
int _min_hits
 
double _min_iou
 
unsigned int _next_id
 
double _nms_iou_thresh
 
bool alive_tracker
 
std::set< int > allItems
 
std::vector< int > assignment
 
std::vector< std::vector< double > > cost_matrix
 
std::vector< int > dead_trackers_id
 
unsigned int detNum = 0
 
std::vector< TrackingBoxframeTrackingResult
 
std::set< int > matchedItems
 
std::vector< cv::Point > matchedPairs
 
double max_centroid_dist_norm = 0.3
 
std::vector< cv::Rect_< float > > predictedBoxes
 
std::vector< KalmanTrackertrackers
 
unsigned int trkNum = 0
 
std::set< int > unmatchedDetections
 
std::set< int > unmatchedTrajectories
 

Detailed Description

Definition at line 33 of file sort.hpp.

Constructor & Destructor Documentation

◆ SortTracker()

SortTracker::SortTracker ( int  max_age = 50,
int  min_hits = 5,
int  max_missed = 7,
double  min_iou = 0.1,
double  nms_iou_thresh = 0.5,
double  min_conf = 0.3 
)

Definition at line 10 of file sort.cpp.

Member Function Documentation

◆ GetCentroidsDistance()

double SortTracker::GetCentroidsDistance ( cv::Rect_< float >  bb_test,
cv::Rect_< float >  bb_gt 
)

Definition at line 35 of file sort.cpp.

◆ GetIOU()

double SortTracker::GetIOU ( cv::Rect_< float >  bb_test,
cv::Rect_< float >  bb_gt 
)
static

Definition at line 23 of file sort.cpp.

Referenced by apply_nms().

◆ update()

void SortTracker::update ( std::vector< cv::Rect >  detection,
int  frame_count,
double  image_diagonal,
std::vector< float >  confidences,
std::vector< int >  classIds 
)

Definition at line 84 of file sort.cpp.

Member Data Documentation

◆ _max_age

int SortTracker::_max_age

Definition at line 63 of file sort.hpp.

◆ _max_missed

int SortTracker::_max_missed

Definition at line 64 of file sort.hpp.

◆ _min_conf

double SortTracker::_min_conf

Definition at line 67 of file sort.hpp.

◆ _min_hits

int SortTracker::_min_hits

Definition at line 62 of file sort.hpp.

◆ _min_iou

double SortTracker::_min_iou

Definition at line 65 of file sort.hpp.

◆ _next_id

unsigned int SortTracker::_next_id

Definition at line 68 of file sort.hpp.

◆ _nms_iou_thresh

double SortTracker::_nms_iou_thresh

Definition at line 66 of file sort.hpp.

◆ alive_tracker

bool SortTracker::alive_tracker

Definition at line 69 of file sort.hpp.

◆ allItems

std::set<int> SortTracker::allItems

Definition at line 53 of file sort.hpp.

◆ assignment

std::vector<int> SortTracker::assignment

Definition at line 50 of file sort.hpp.

◆ cost_matrix

std::vector<std::vector<double> > SortTracker::cost_matrix

Definition at line 49 of file sort.hpp.

◆ dead_trackers_id

std::vector<int> SortTracker::dead_trackers_id

Definition at line 58 of file sort.hpp.

◆ detNum

unsigned int SortTracker::detNum = 0

Definition at line 61 of file sort.hpp.

◆ frameTrackingResult

std::vector<TrackingBox> SortTracker::frameTrackingResult

Definition at line 57 of file sort.hpp.

◆ matchedItems

std::set<int> SortTracker::matchedItems

Definition at line 54 of file sort.hpp.

◆ matchedPairs

std::vector<cv::Point> SortTracker::matchedPairs

Definition at line 55 of file sort.hpp.

◆ max_centroid_dist_norm

double SortTracker::max_centroid_dist_norm = 0.3

Definition at line 46 of file sort.hpp.

◆ predictedBoxes

std::vector<cv::Rect_<float> > SortTracker::predictedBoxes

Definition at line 48 of file sort.hpp.

◆ trackers

std::vector<KalmanTracker> SortTracker::trackers

Definition at line 44 of file sort.hpp.

◆ trkNum

unsigned int SortTracker::trkNum = 0

Definition at line 60 of file sort.hpp.

◆ unmatchedDetections

std::set<int> SortTracker::unmatchedDetections

Definition at line 51 of file sort.hpp.

◆ unmatchedTrajectories

std::set<int> SortTracker::unmatchedTrajectories

Definition at line 52 of file sort.hpp.


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