Skip to content

deeplabcut.pose_estimation_pytorch.data

Modules:

Name Description
base
cocoloader
collate

Custom collate functions.

ctd
dataset
dlcloader

Class implementing the Loader for DeepLabCut projects.

generative_sampling

A file containing code to perform generative sampling of keypoints for CTD.

helper
image

Classes and functions to manipulate images.

postprocessor

Post-process predictions made by models.

preprocessor

Helpers to run preprocess data before running inference.

snapshots

Code to handle storing models.

transforms
utils