wormpose.config.experiment_config

class wormpose.config.experiment_config.ExperimentConfig(num_train_samples: Optional[int] = None, num_eval_samples: Optional[int] = None, image_shape: Optional[Tuple[int, int]] = None, dataset_loader: Optional[str] = None, theta_dimensions: Optional[int] = None, resize_factor: float = 1.0, video_names: Optional[List[str]] = None, worm_is_lighter: bool = False)

Data container for the experiment config, created when generating training data

wormpose.config.experiment_config.add_config_argument(parser)

For command line arguments, add the option to pass the path of the configuration file

Parameters

parser – Argparse parser

wormpose.config.experiment_config.load_config(config_filepath: str) → wormpose.config.experiment_config.ExperimentConfig

Load the experiment config from a json file

Parameters

config_filepath – path of the config json file to load

Returns

loaded config object

wormpose.config.experiment_config.save_config(experiment_config: wormpose.config.experiment_config.ExperimentConfig, config_filepath: str)

Save the experiment config to a json file

Parameters

• experiment_config – config object to save

• config_filepath – path where to write the config json file