Class Task

deeplabcut/pose_estimation_pytorch/task.py:25–43

A task to solve.

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}"