BaseSelector

BaseSelector inheritance diagram

class savant.base.selector.BaseSelector(**kwargs)

Selector interface.

abstract __call__(bbox_tensor)

Filters objects.

Parameters:

bbox_tensor (ndarray) – Bounding boxes for selection, represented as numpy array and contains (class_id, confidence, xc, yc, width, height, [angle])

Returns:

Resulting BBox tensor which contains only the selected boxes

Return type:

ndarray