wormpose.dataset.loaders.tierpsy.frames_dataset

Implementation of BaseFramesDataset to load Tierpsy tracker frames

class wormpose.dataset.loaders.tierpsy.frames_dataset.FramesDataset(dataset_path: str)

open(video_name: str)

The frames of the dataset are accessed trough a context manager object, in this way we have the option of not entirely loading a big image array in memory if possible

Parameters

video_name – One video unique id (should be one value of video_names())

Returns

A context manager object that can be used with the “with” python statement giving access to the frames (array of images) of the dataset Example use : frames = open(“video0”)

video_names()

A dataset is composed of several videos

Returns

A list of unique ids (string) identifying a video in the dataset