annotations_to_heatmap

hybrid_learning.datasets.custom.coco.keypoints_processing.annotations_to_heatmap(annotation_wh, annotations, keypoint_idxs, skeleton, pt_radius=0.025, link_width=None)

Create a heatmap marking the centroids of the linked keypoints from the annotations list. Keypoints are specified by their keypoint_idxs. For details on the parameters, see annotations_to_mask().

Parameters

• pt_radius (float) – this is used as standard deviation and as the point radius to calculate connected components

• annotation_wh (Tuple[int, int]) –

• annotations (List[dict]) –

• keypoint_idxs (Union[Sequence[int], Iterable[Iterable[int]]]) –

• skeleton (Sequence[Tuple[int, int]]) –

• link_width (Optional[float]) –

Returns

heatmap (1-channel grayscale image in mode 'L') of size annotation_wh

Return type

Image