MCPcopy Create free account
hub / github.com/DeepLabCut/DeepLabCut / load_demo_data

Function load_demo_data

deeplabcut/create_project/demo_data.py:20–50  ·  view source on GitHub ↗

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.

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

Source from the content-addressed store, hash-verified

18
19
20def load_demo_data(
21 config: str,
22 createtrainingset: bool = True,
23 engine: Engine = Engine.PYTORCH,
24):
25 """Loads the demo data -- subset from trail-tracking data in Mathis et al. 2018.
26 When loading, it sets paths correctly to run this project on your system.
27
28 Parameter
29 ----------
30 config : string
31 Full path of the config.yaml file of the provided demo dataset as a string.
32
33 createtrainingset : bool
34 Boolean variable indicating if a training set shall be created.
35
36 engine: Engine
37 The Engine to create the training set for if a training set shall be created.
38
39 Example
40 --------
41 >>> deeplabcut.load_demo_data('config.yaml')
42 --------
43 """
44 config = Path(config).resolve()
45 config = str(config)
46
47 transform_data(config)
48 if createtrainingset:
49 print("Loaded, now creating training data...")
50 deeplabcut.create_training_dataset(config, num_shuffles=1, engine=engine)
51
52
53def transform_data(config):

Callers

nothing calls this directly

Calls 2

transform_dataFunction · 0.85

Tested by

no test coverage detected