BBoxSelector

Inheritance diagram of BBoxSelector

BBoxSelector inheritance diagram

class savant.selector.detector.BBoxSelector(confidence_threshold=0.5, nms_iou_threshold=0.5, top_k=0, min_width=0, min_height=0, max_width=0, max_height=0, **kwargs)

Detector bbox per class selector.

Parameters:
  • confidence_threshold (float) – confidence threshold

  • nms_iou_threshold (float) – nms iou threshold

  • top_k (int) – top k bboxes to keep

  • min_width (int) – minimal bbox width

  • min_height (int) – minimal bbox height

  • max_width (int) – maximum bbox width

  • max_height (int) – maximum bbox height

__call__(bbox_tensor)

Filters bboxes by confidence and size, applies NMS.

Parameters:

bbox_tensor (ndarray) – tensor(class_id, confidence, left, top, width, height)

Returns:

Selected bbox tensor

Return type:

ndarray