wormpose.commands.generate_training_data

Generates the training and evaluation data from a dataset.

wormpose.commands.generate_training_data.generate(dataset_loader: str, dataset_path: str, **kwargs)[source]

Generate synthetic images (training data) and processed real images (evaluation data) and save them to TFrecord files using multiprocessing

Parameters
  • dataset_loader – Name of the dataset loader, for example “tierpsy”

  • dataset_path – Root path of the dataset containing videos of worm