API reference

deepposekit.io

Data loading primitives: annotation datasets, training generators and video readers.

DataGenerator

Reads an annotation HDF5 file and exposes images + keypoints.

DataGenerator(datapath, dataset='images', mode='full', generator=None)

TrainingGenerator

Wraps a DataGenerator with augmentation, downsampling and Gaussian confidence maps.

TrainingGenerator(
    generator,
    downsample_factor=2,
    use_graph=True,
    augmenter=None,
    shuffle=True,
    sigma=5,
    validation_split=0.1,
    graph_scale=1.0,
    random_seed=None,
)

VideoReader

Streams video frames in batches, ready to feed straight into model.predict().

VideoReader(filepath, batch_size=1, frame_size=None, gray=False)

initialize_dataset

Build a fresh annotation HDF5 file from a folder of images and a skeleton definition.

initialize_dataset(images, datapath, skeleton, overwrite=False)