deeplabcut.pose_estimation_pytorch.task

Types of tasks that can be run by DeepLabCut pose estimation models.

Classes:

Name	Description
Task	A task to solve.

Task dataclass

Bases: TaskDataMixin, Enum

A task to solve.

Source code in deeplabcut/pose_estimation_pytorch/task.py

class Task(TaskDataMixin, Enum):
    """A task to solve."""

    BOTTOM_UP = ("BU", "BottomUp"), "snapshot"
    DETECT = ("DT", "Detect"), "snapshot-detector"
    TOP_DOWN = ("TD", "TopDown"), "snapshot"
    COND_TOP_DOWN = ("CTD", "CondTopDown", "ConditionalTopDown"), "snapshot"

    @classmethod
    def _missing_(cls, value):
        if isinstance(value, str):
            value = value.upper()
            for member in cls:
                if value in member.aliases:
                    return member
        return None

    def __repr__(self) -> str:
        return f"Task.{self.name}"