tracklab.wrappers.datasets package

Subpackages

Submodules

tracklab.wrappers.datasets.external_video module

class tracklab.wrappers.datasets.external_video.ExternalVideo(dataset_path: str, video_path: str, *args, **kwargs)

Bases: TrackingDataset

A class to use PbTrack at inference on any .mp4 video. The .mp4 image frames are first saved individually to disk in ‘working_dir/tmp/’ dir. A tracking test set is then created based on these images files, with video_metadata and image_metadata, but without detections. TODO dirty/easy implementation, we should refactor the tracking engine to make it load directly the images from the

mp4. Each video in video_metadata should specify its type: folder of images or mp4 video or youtube link, etc, and tracking engine should adapt its batch loop accordingly.

annotations_dir = 'posetrack_data'

static get_frame_count(video_path)

tracklab.wrappers.datasets.external_video.write_video_images_to_disk(video_path)

tracklab.wrappers.datasets.jrdb_pose module

tracklab.wrappers.datasets.jta module