wormpose.machine_learning.tfrecord_file

Functions to read and write TFrecord files containing wormpose labeled data:

one image paired with centerline angles (two options for head-tail flip)

class wormpose.machine_learning.tfrecord_file.TfrecordLabeledDataWriter(filename: str)

wormpose.machine_learning.tfrecord_file.read(filename: str, theta_dims: int)

Read a tfrecord file where the images in the files have the same width and height