Skip to content

deeplabcut.create_project.demo_data

Functions:

Name Description
load_demo_data

Loads the demo data -- subset from trail-tracking data in Mathis et al. 2018.
transform_data

This function adds the full path to labeling dataset.

load_demo_data 

load_demo_data(config: str, createtrainingset: bool = True, engine: Engine = Engine.PYTORCH)

Loads the demo data -- subset from trail-tracking data in Mathis et al. 2018. When loading, it sets paths correctly to run this project on your system.

Parameter

config : string Full path of the config.yaml file of the provided demo dataset as a string.

createtrainingset : bool Boolean variable indicating if a training set shall be created.

engine: Engine The Engine to create the training set for if a training set shall be created.

Example

deeplabcut.load_demo_data('config.yaml')

Source code in deeplabcut/create_project/demo_data.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
def load_demo_data(
    config: str,
    createtrainingset: bool = True,
    engine: Engine = Engine.PYTORCH,
):
    """Loads the demo data -- subset from trail-tracking data in Mathis et al. 2018.
    When loading, it sets paths correctly to run this project on your system.

    Parameter
      ----------
      config : string
          Full path of the config.yaml file of the provided demo dataset as a string.

      createtrainingset : bool
          Boolean variable indicating if a training set shall be created.

      engine: Engine
          The Engine to create the training set for if a training set shall be created.

      Example
      --------
      >>> deeplabcut.load_demo_data('config.yaml')
      --------
    """
    config = Path(config).resolve()
    config = str(config)

    transform_data(config)
    if createtrainingset:
        print("Loaded, now creating training data...")
        deeplabcut.create_training_dataset(config, num_shuffles=1, engine=engine)

transform_data 

transform_data(config)

This function adds the full path to labeling dataset.

It also adds the correct path to the video file in the config file.

Source code in deeplabcut/create_project/demo_data.py 
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
def transform_data(config):
    """This function adds the full path to labeling dataset.

    It also adds the correct path to the video file in the config file.
    """

    cfg = auxiliaryfunctions.read_config(config)
    project_path = str(Path(config).parents[0])

    cfg["project_path"] = project_path
    if "Reaching" in project_path:
        video_file = os.path.join(project_path, "videos", "reachingvideo1.avi")
    elif "openfield" in project_path:
        video_file = os.path.join(project_path, "videos", "m4s1.mp4")
    else:
        print("This is not an official demo dataset.")

    if "WILL BE AUTOMATICALLY UPDATED BY DEMO CODE" in cfg["video_sets"].keys():
        cfg["video_sets"][str(video_file)] = cfg["video_sets"].pop("WILL BE AUTOMATICALLY UPDATED BY DEMO CODE")

    auxiliaryfunctions.write_config(config, cfg)