Skip to content

deeplabcut.modelzoo.generalized_data_converter.datasets.materialize

Functions:

Name Description
modelzoo_ma_template

Default ProjectConfig for generalized_data_converter multi-animal projects.

modelzoo_single_template

Default ProjectConfig for generalized_data_converter single-animal projects.

modelzoo_ma_template

modelzoo_ma_template() -> ProjectConfig

Default ProjectConfig for generalized_data_converter multi-animal projects.

Source code in deeplabcut/modelzoo/generalized_data_converter/datasets/materialize.py
def modelzoo_ma_template() -> ProjectConfig:
    """Default ProjectConfig for generalized_data_converter multi-animal projects."""
    modelzoo_single_template()

    return modelzoo_single_template().model_copy(
        update={
            "default_augmenter": "multi-animal-imgaug",
            "croppedtraining": True,
            "multianimalproject": True,
            "individuals": [],
            "multianimalbodyparts": [],
            "bodyparts": "MULTI!",
        }
    )

modelzoo_single_template

modelzoo_single_template() -> ProjectConfig

Default ProjectConfig for generalized_data_converter single-animal projects.

Source code in deeplabcut/modelzoo/generalized_data_converter/datasets/materialize.py
def modelzoo_single_template() -> ProjectConfig:
    """Default ProjectConfig for generalized_data_converter single-animal projects."""
    return ProjectConfig(
        Task="",
        project_path=Path(),
        scorer="",
        date="",
        video_sets={},
        skeleton=[],
        bodyparts=[],
        start=0,
        stop=1,
        numframes2pick=42,
        skeleton_color="black",
        pcutoff=0.6,
        dotsize=8,
        alphavalue=0.7,
        colormap="rainbow",
        TrainingFraction=[],
        iteration=0,
        default_net_type="resnet_50",
        default_augmenter="imgaug",
        snapshotindex=-1,
        batch_size=8,
        cropping=False,
        croppedtraining=False,
        multianimalproject=False,
        uniquebodyparts=[],
        x1=0,
        x2=640,
        y1=277,
        y2=624,
        corner2move2=[50, 50],
        move2corner=True,
        identity=False,
    )