Skip to content

deeplabcut.pose_estimation_pytorch.modelzoo

Public API for PyTorch modelzoo.

Exports are resolved lazily to avoid import cycles between helpers and package initialization.

Modules:

Name Description
config

Methods to create the configuration files to fine-tune SuperAnimal models.

fmpose_3d

DeepLabCut2.0-3.0 Toolbox (deeplabcut.org)

inference
inference_helpers

PyTorch-specific helper entrypoints for model zoo inference.

memory_replay
train_from_coco

File to train a model on a COCO dataset.

utils