tracklab.wrappers.dataset package

Subpackages

Submodules

tracklab.wrappers.dataset.external_video module

class tracklab.wrappers.dataset.external_video.ExternalVideo(dataset_path: str, video_path: str, *args, **kwargs)[source]

Bases: TrackingDataset

A class to use Tracklab 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.

static get_frame_count(video_path)[source]
tracklab.wrappers.dataset.external_video.write_video_images_to_disk(video_path)[source]

tracklab.wrappers.dataset.jrdb_pose module

tracklab.wrappers.dataset.jta module