fiftyone.utils.torch.KeypointDetectorOutputProcessor

class documentation

class KeypointDetectorOutputProcessor(OutputProcessor): (source)

Constructor: KeypointDetectorOutputProcessor(classes)

View In Hierarchy

Output processor for keypoint detection models.

Parameters
classesthe list of class labels for the model
Method __call__ Parses the model output.
Method __init__ Undocumented
Instance Variable classes Undocumented
Method _parse_output Undocumented
def __call__(self, output, frame_size, confidence_thresh=None): (source)
overrides fiftyone.utils.torch.OutputProcessor.__call__

Parses the model output.

Parameters
output

a batch of predictions output = List[Dict[Tensor]], where each dict has the following keys:

  • boxes (FloatTensor[N, 4]): the predicted boxes in [x1, y1, x2, y2] format (absolute coordinates)
  • labels (Int64Tensor[N]): the predicted labels
  • scores (Tensor[N]): the scores for each prediction
  • keypoints (FloatTensor[N, K, ...]): the predicted keypoints for each instance in [x, y, v, ...] format
frame_sizethe (width, height) of the frames in the batch
confidence_thresh:Nonean optional confidence threshold to use to filter any applicable predictions
Returns
a list of fiftyone.core.labels.Label dicts
def __init__(self, classes=None): (source)
overrides fiftyone.utils.torch.OutputProcessor.__init__

Undocumented

classes: None = (source)
overrides fiftyone.utils.torch.OutputProcessor.classes

Undocumented

def _parse_output(self, output, frame_size, confidence_thresh): (source)

Undocumented