wormpose.commands.train_model

Trains the neural network on the training data, supports resuming training

wormpose.commands.train_model.train(dataset_path: str, **kwargs)

Train a neural network with the TFrecord files generated with the script generate_training_data Save the best model performing on evaluation data

Parameters

dataset_path – Root path of the dataset containing videos of worm