babysister.sort_wrapper module

class babysister.sort_wrapper.SORT

Bases: object

Wrapper for sort.Sort.

update(boxes, scores)

Update tracked objects.

Parameters:

• boxes (list) – Boxes coordinate in format [[x0, y0, x1, y1], …].
• scores (list) – Confidence scores.

Returns:

detections in the format [[x0, y0, x1, y1, id], …]

Return type:

ndarray

Requires:

This method must be called once for each frame even with empty detections.

Note

The number of objects returned may differ from the number of boxes provided.