def convert_to_example(pose, shape=None): """Build an Example proto for an image example. Args: pose: 72-D vector, float shape: 10-D vector, float Returns: Example proto """ if shape is None: example = tf.train.Example(features=tf.train.Features( feature={'pose': float_feature(pose.astype(np.float))})) else: example = tf.train.Example(features=tf.train.Features( feature={ 'pose': float_feature(pose.astype(np.float)), 'shape': float_feature(shape.astype(np.float)), })) return example
def convert_to_example_temporal(poses): """ Builds example proto for a changes in pose. Args: pose (50x24x3): Axis-angle representation of the current pose. Returns: Example proto. """ return tf.train.Example(features=tf.train.Features( feature={ 'pose': float_feature(poses.ravel()), }))