sort

README

sort-tracking

SORT algorithm for online object tracking in videos

Highly inpired on https://github.com/abewley/sort/blob/master/sort.py

Documentation

Index

• type KalmanBoxTracker
  • func NewKalmanBoxTracker(bbox []float64) KalmanBoxTracker
• type Sort
  • func NewSort(maxAge int, minHits int) Sort
  • func (s Sort) Update(dets [][]float64, iouThreshold float64)

Types

type KalmanBoxTracker

type KalmanBoxTracker struct {
	// contains filtered or unexported fields
}

KalmanBoxTracker This class represents the internel state of individual tracked objects observed as bbox.

func NewKalmanBoxTracker

func NewKalmanBoxTracker(bbox []float64) KalmanBoxTracker

NewKalmanBoxTracker Initialises a tracker using initial bounding box.

type Sort

type Sort struct {
	// contains filtered or unexported fields
}

Sort Detection tracking

func NewSort

func NewSort(maxAge int, minHits int) Sort

NewSort initializes a new SORT tracking session

func (Sort) Update

func (s Sort) Update(dets [][]float64, iouThreshold float64)

Update update trackers from detections

Params:
  dets - a numpy array of detections in the format [[x1,y1,x2,y2,score],[x1,y1,x2,y2,score],...]
Requires: this method must be called once for each frame even with empty detections.
Returns the a similar array, where the last column is the object ID.
NOTE: The number of objects returned may differ from the number of detections provided.