LabeledData

class lightning_pose.utils.io.LabeledData

Bases: object

Result of parsing a label CSV file.

keypoint_names

ordered list of keypoint name strings

Type:

list[str]

image_names

ordered list of image path strings (relative to the project root)

Type:

list[str]

keypoints

(N, K, 2) float tensor of (x, y) coordinates; NaN where unlabeled

Type:

jaxtyping.Float[Tensor, ‘num_frames num_keypoints 2’]

visibility

(N, K) int64 tensor of per-keypoint visibility flags (0, 1, or 2), or None when the CSV does not contain a visible column

Type:

jaxtyping.Int[Tensor, ‘num_frames num_keypoints’] | None

Attributes Summary

image_names
keypoint_names
keypoints
visibility

Attributes Documentation

image_names: list[str] = <dataclasses._MISSING_TYPE object>

keypoint_names: list[str] = <dataclasses._MISSING_TYPE object>

keypoints: Float[Tensor, 'num_frames num_keypoints 2'] = <dataclasses._MISSING_TYPE object>

visibility: Int[Tensor, 'num_frames num_keypoints'] | None = <dataclasses._MISSING_TYPE object>

__init__(keypoint_names: list[str], image_names: list[str], keypoints: Float[Tensor, 'num_frames num_keypoints 2'], visibility: Int[Tensor, 'num_frames num_keypoints'] | None) → None

__new__(**kwargs)