BaseTrackingDataset

class lightning_pose.data.datasets.BaseTrackingDataset(root_directory: str, csv_path: str, header_rows: List[int] | None = [0, 1, 2], imgaug_transform: Callable | None = None, do_context: bool = False)

Bases: Dataset

Base dataset that contains images and keypoints as (x, y) pairs.

Attributes Summary

height
width

Attributes Documentation

height

width